2016-05-31 15:56:57
题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2809
直观的思想是当领导力确定时,尽量选择薪水少的---->枚举领导力
树上每个节点维护一个大根堆,先将所有儿子并起来,若超过预算,再弹出根直到满足预算。
左偏树的合并操作是logn的,遍历一遍树,时间复杂度O(nlogn)
1 #include<bits/stdc++.h> 2 #define inf 1000000000 3 #define ll long long 4 #define N 1000005 5 using namespace std; 6 int read(){ 7 int x=0,f=1;char ch=getchar(); 8 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 9 while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} 10 return x*f; 11 } 12 struct Node{ 13 int to,next; 14 }e[N]; 15 int n,m,tot,head[N],v[N],lead[N],dep[N],ls[N],rs[N],sz[N],rt[N]; 16 ll sum[N],ans; 17 void add(int x,int y){ 18 e[++tot]=(Node){y,head[x]};head[x]=tot; 19 } 20 int merge(int x,int y){ 21 if(!x||!y)return x+y; 22 if(v[x]<v[y])swap(x,y); 23 rs[x]=merge(rs[x],y); 24 if(dep[rs[x]]>dep[ls[x]])swap(rs[x],ls[x]); 25 dep[x]=dep[rs[x]]+1; 26 return x; 27 } 28 void dfs(int x){ 29 sz[x]=1;sum[x]=1ll*v[x]; 30 for(int i=head[x];i;i=e[i].next){ 31 dfs(e[i].to); 32 sz[x]+=sz[e[i].to]; 33 sum[x]+=sum[e[i].to]; 34 rt[x]=merge(rt[x],rt[e[i].to]); 35 } 36 while(sum[x]>m){ 37 sum[x]-=1ll*v[rt[x]];rt[x]=merge(ls[rt[x]],rs[rt[x]]); 38 sz[x]--; 39 } 40 ans=max(ans,1ll*sz[x]*lead[x]); 41 } 42 int main(){ 43 n=read();m=read(); 44 for(int i=1;i<=n;i++){ 45 add(read(),i);v[i]=read();lead[i]=read();rt[i]=i; 46 } 47 dfs(0); 48 printf("%d\n",ans); 49 return 0; 50 }