[BZOJ2809][Apio2012]dispatching（可并堆）

=== ===

这里放传送门

=== ===

题解

可以看出因为读入的东西是一棵树，并且选定一个管理者以后派遣的忍者肯定都在它的子树里面，所以可以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和随时更新堆的根结点指针不然指针会指到奇怪的地方去