洛谷p1552[APIO2012] 派遣（树形dp+左偏树）

题意：master可以向所有被派遣的忍者发送指令，master自己可以被派遣，也可以不派遣，如果master没有派遣，就不需要支付master的薪水。上级给直属下级发送消息。每个忍者的上级bi，薪水ci，领导力li，满意度为派遣人数×master的领导力水平，输出满意度最大值

分析：用siz记录派遣人数，薪水总数sum，用树形dp，如果薪水总数超过预期数m，可以把让根节点不去，用左偏树删除。

代码：

#include<bits/stdc++.h>
#define maxn 200010
using namespace std;
typedef long long ll;
ll n,m,root,ans;
ll fa[maxn],ls[maxn],rs[maxn],dis[maxn],val[maxn],l[maxn],sum[maxn],siz[maxn];
struct edge
{
    int to,nxt;
}e[maxn];
int head[maxn],edge_cnt;
void add(int from,int to)
{
    e[++edge_cnt]=(edge){to,head[from]};
    head[from]=edge_cnt;
}
int find(int x)
{
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
int merge(int x,int y)
{
	if(x==y) return 0;
    if(!x||!y) return x+y;
	if(val[x]<val[y]) swap(x,y);
	rs[x]=merge(rs[x],y),fa[rs[x]]=x;
	if(dis[ls[x]]<dis[rs[x]]) swap(ls[x],rs[x]);
	if(rs[x]) dis[x]=dis[rs[x]]+1;
	else dis[x]=0;
	return x;
}
void del(int x)
{
	fa[ls[x]]=ls[x],fa[rs[x]]=rs[x];
	fa[x]=merge(ls[x],rs[x]);
}
void dfs(int x)
{
    for(int i=head[x];i;i=e[i].nxt)
    {
        int y=e[i].to;//儿子
        dfs(y);
        siz[x]+=siz[y];//当前长度加上儿子的长度
        sum[x]+=sum[y];//加上薪水
        merge(find(x),find(y));//合并子树
    }
    while(sum[x]>m)//如果薪水总和大于m
    {
        int rootx=find(x);//根
        siz[x]--;//减少子树
        sum[x]-=val[rootx];//减少子树总和
        del(rootx);//删除根
    }
    ans=max(ans,l[x]*siz[x]);
}
int main()
{
	cin>>n>>m;
    for(int i=1;i<=n;++i){
        int fath;
        cin>>fath>>val[i]>>l[i];
        sum[i]=val[i],siz[i]=1,fa[i]=i;
        if(fath) add(fath,i);//建立边
        else root=i;
    }
    dfs(root);
    cout<<ans<<endl;
	return 0;
}