=== ===
这里放传送门
=== ===
题解
可以看出因为读入的东西是一棵树,并且选定一个管理者以后派遣的忍者肯定都在它的子树里面,所以可以dfs来求解。一开始的思路是按薪水要求维护小根堆然后每次把最小的往外拿,但这样合并到父节点的时候可能要把它们再塞回去,这显然不大对。。。正确的做法应该维护大根堆,然后对每个节点附加一个sum域表示堆中它的子树的权值之和,当这个和大于m的时候就把最大的元素往外拿,这样弹出的是最少的,留下的就是最多的,并且这些弹出去的元素以后都不会再塞回来。删完了以后这可子树的答案就是堆中留下的元素即它的size乘以子树根节点的管理能力值。合并两棵子树的过程是可并堆最基本的操作,这里使用左偏树来实现可并堆,合并的过程merge中带入两个需要合并的堆的堆顶指针x和y,如果两个堆中有一个为空就返回另一个,否则选择两个根节点较小的一个当作新的根节点,然后把另一个堆合并到它的右子树上去,递归进行。递归出来把左右子树中NPL值较小的一个交换到右边,这样再把别的东西挂上去的时候就不会让树倾斜得太厉害。删除根节点也可以通过merge操作进行,合并左右子树即可。注意维护NPL(null path length)域,它是左偏树最坏情况下时间复杂度保持 O(nlogn) 的关键。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,l[100010],p[100010],a[200010],next[200010],tot;
long long ans;
struct heap{
heap *l,*r;
int val,NPL,size;
long long sum;
heap();
void count(){
NPL=r->NPL+1;
size=l->size+r->size+1;
sum=(long long)l->sum+r->sum+val;
}
}*null,H[100010],*h[100010];
heap::heap(){
l=r=null;val=NPL=sum=0;size=0;
}
heap *merge(heap *x,heap *y){
if (x==null) return y;
if (y==null) return x;
if (x->val<y->val) swap(x,y);
x->r=merge(x->r,y);
if (x->r->NPL>x->l->NPL) swap(x->r,x->l);
x->count();//更新操作
return x;
}
void add(int x,int y){
tot++;a[tot]=y;next[tot]=p[x];p[x]=tot;
}
void search(int u,int fa){
for (int i=p[u];i!=0;i=next[i])
if (a[i]!=fa){
search(a[i],u);
h[u]=merge(h[u],h[a[i]]);//注意更新根节点
}
while (h[u]->sum>m)
h[u]=merge(h[u]->l,h[u]->r);//删除堆顶元素
ans=max(ans,(long long)h[u]->size*l[u]);
}
int main()
{
null=new heap;
*null=heap();
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++){H[i]=heap();h[i]=H+i;}
for (int i=1;i<=n;i++){
int B,C;
scanf("%d%d%d",&B,&C,&l[i]);
if (B!=0){add(i,B);add(B,i);}
H[i].val=H[i].sum=C;H[i].size=1;
}
search(1,0);
printf("%I64d\n",ans);
return 0;
}
偏偏在最后出现的补充说明
注意强转long long和随时更新堆的根结点指针不然指针会指到奇怪的地方去