【洛谷P1552】【APIO2012】派遣

题目背景

在一个忍者的帮派里，一些忍者们被选中派遣给顾客，然后依据自己的工作获取报偿。

题目描述

在这个帮派里，有一名忍者被称之为Master。除了Master以外，每名忍者都有且仅有一个上级。为保密，同时增强忍者们的领导力，所有与他们工作相关的指令总是由上级发送给他的直接下属，而不允许通过其他的方式发送。

现在你要招募一批忍者，并把它们派遣给顾客。你需要为每个被派遣的忍者支付一定的薪水，同时使得支付的薪水总额不超过你的预算。另外，为了发送指令，你需要选择一名忍者作为管理者，要求这个管理者可以向所有被派遣的忍者发送指令，在发送指令时，任何忍者（不管是否被派遣）都可以作为消息的传递人。管理者自己可以被派遣，也可以不被派遣。当然，如果管理者没有被排遣，你就不需要支付管理者的薪水。

你的目标是在预算内使顾客的满意度最大。这里定义顾客的满意度为派遣的忍者总数乘以管理者的领导力水平，其中每个忍者的领导力水平也是一定的。

写一个程序，给定每一个忍者i的上级Bi，薪水Ci，领导力Li，以及支付给忍者们的薪水总预算M，输出在预算内满足上述要求时顾客满意度的最大值。

输入输出格式

输入格式：
第一行包含两个整数N和M，其中N表示忍者的个数，M表示薪水的总预算。

接下来N行描述忍者们的上级、薪水以及领导力。其中的第i行包含三个整数Bi,Ci,Li分别表示第i个忍者的上级，薪水以及领导力。Master满足Bi=0，并且每一个忍者的老板的编号一定小于自己的编号Bi

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define N 100005
#define ll long long
int val[N],dis[N],ret[N],Next[N],Head[N];
int ls[N],rs[N],root[N],num[N],sum[N];
int c[N],l[N];
ll ans;
int n,m,tot,cnt;
using namespace std;
inline int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
inline void insert(int u,int v)
{
    tot++;
    ret[tot]=v;
    Next[tot]=Head[u];
    Head[u]=tot;
}
int merge(int a,int b)
{
    if (!a) return b;
    if (!b) return a;
    if (val[a]<val[b]) swap(a,b);
    rs[a]=merge(rs[a],b);
    if (dis[ls[a]]<dis[rs[a]]) swap(ls[a],rs[a]);
    dis[a]=dis[rs[a]]+1;
    return a;
}
int main()
{
    n=read();m=read();dis[0]=-1;
    for (int i=1;i<=n;i++)
    {
        int b=read();c[i]=read();l[i]=read();
        insert(b,i);
    }
    for (int u=n;u>=1;u--)
    {
        if (c[u]<=m)
        {
            cnt++;
            root[u]=cnt;
            val[cnt]=c[u];
            sum[u]=c[u];
            num[u]=1;
        }
        for (int i=Head[u];i;i=Next[i])
        {
            int v=ret[i];
            root[u]=merge(root[u],root[v]);
            num[u]+=num[v];
            sum[u]+=sum[v];
            while (sum[u]>m)
            {
                num[u]--;
                sum[u]-=val[root[u]];
                root[u]=merge(ls[root[u]],rs[root[u]]);
            }
        }
        ans=max(ans,(ll)l[u]*num[u]);
    }
    printf("%lld",ans);
    return 0;
}