2809: [Apio2012]dispatching

题目链接

题目大意：n个点组成一棵树，每个点都有一个领导力和费用，可以让一个点当领导，然后在这个点的子树中选择一些费用之和不超过m的点，得到领导的领导力乘选择的点的个数（领导可不被选择）的利润。求利润最大值

题解：主席树/启发式合并可做
当然是选择可并堆啦
先选择所有点，超了就把最大的扔掉，用堆维护
因为每个点的堆由其子树的堆合并得到，需要可并堆

我的收获：堆维护的套路贪心

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define M 222222

int n,p,t;
int root[M],c[M][2],size[M],head[M];
long long m,ans;
long long k[M],l[M],sum[M];
bool vis[M];

struct edge{int to,nex;}e[M*4];

void add(int u,int v){e[t]=(edge){v,head[u]};head[u]=t++;}

int merge(int x,int y){
    if(!x||!y) return x+y;
    if(k[x]<k[y]) swap(x,y);
    c[x][1]=merge(c[x][1],y);
    root[c[x][1]]=x;
    swap(c[x][0],c[x][1]);
    return x;
}

void pop(int &x){x=merge(c[x][0],c[x][1]);}

void dfs(int x){
    sum[x]=k[x];size[x]=1;
    for(int v,i=head[x];i!=-1;i=e[i].nex){
        dfs(v=e[i].to);
        sum[x]+=sum[v];
        size[x]+=size[v];
        root[x]=merge(root[v],root[x]);
    }
    while(sum[x]>m) sum[x]-=k[root[x]],pop(root[x]),size[x]--;
    if(size[x]*l[x]>ans) ans=size[x]*l[x];
}

void work()
{
    int i;
    for(i=1;i<=n;i++) if(!vis[i]) break;//找到根进行dfs 
    dfs(i);
    printf("%lld",ans);
}

void init()
{
    cin>>n>>m;t=0;memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++){
        scanf("%d%lld%lld",&p,&k[i],&l[i]);
        if(p) add(p,i),vis[i]=1;
        root[i]=i;c[i][0]=c[i][1]=0;
    }
}

int main()
{
    init();
    work();
    return 0;
}