A - dispatching
在一个忍者的帮派里,一些忍者们被选中派遣给顾客,然后依据自己的工作获取报偿。在这个帮派里,有一名忍者被称之为 Master。除了 Master以外,每名忍者都有且仅有一个上级。为保密,同时增强忍者们的领导力,所有与他们工作相关的指令总是由上级发送给他的直接下属,而不允许通过其他的方式发送。现在你要招募一批忍者,并把它们派遣给顾客。你需要为每个被派遣的忍者 支付一定的薪水,同时使得支付的薪水总额不超过你的预算。另外,为了发送指令,你需要选择一名忍者作为管理者,要求这个管理者可以向所有被派遣的忍者 发送指令,在发送指令时,任何忍者(不管是否被派遣)都可以作为消息的传递 人。管理者自己可以被派遣,也可以不被派遣。当然,如果管理者没有被排遣,就不需要支付管理者的薪水。你的目标是在预算内使顾客的满意度最大。这里定义顾客的满意度为派遣的忍者总数乘以管理者的领导力水平,其中每个忍者的领导力水平也是一定的。写一个程序,给定每一个忍者 i的上级 Bi,薪水Ci,领导力L i,以及支付给忍者们的薪水总预算 M,输出在预算内满足上述要求时顾客满意度的最大值。
1 ≤N ≤ 100,000 忍者的个数;
1 ≤M ≤ 1,000,000,000 薪水总预算;
0 ≤B i < i 忍者的上级的编号;
1 ≤Ci ≤ M 忍者的薪水;
1 ≤Li ≤ 1,000,000,000 忍者的领导力水平。
Input
从标准输入读入数据。
第一行包含两个整数 N和 M,其中 N表示忍者的个数,M表示薪水的总预 算。
接下来 N行描述忍者们的上级、薪水以及领导力。其中的第 i 行包含三个整 B i , C i , L i分别表示第i个忍者的上级,薪水以及领导力。Master满足B i = 0, 并且每一个忍者的老板的编号一定小于自己的编号 B i < i。
Output
输出一个数,表示在预算内顾客的满意度的最大值。
Sample Input
5 4
0 3 3
1 3 5
2 2 2
1 2 4
2 3 1
Sample Output
6
Hint
如果我们选择编号为 1的忍者作为管理者并且派遣第三个和第四个忍者,薪水总和为 4,没有超过总预算 4。因为派遣了2个忍者并且管理者的领导力为3,用户的满意度为 2,是可以得到的用户满意度的最大值。
在每个结点上都按薪水建立一个大顶堆,并记录堆中结点的数量和sum,如果sum>m则贪心去除堆顶,向上遍历时合并,每一个结点上都会有一个最大值,答案就是这些最大值中的最大值。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e5+5;
int f[N]; // 父节点
vector<int> v[N];
struct Tree //定义节点
{
int val,ch[2],d;
}t[N];
int lead[N];
int& rs(int x)//求右儿子
{
return t[x].ch[t[t[x].ch[1]].d<t[t[x].ch[0]].d];//为了满足左偏树的性质2,须保证右子树节点的dist小于左子树节点的dist
}
int merge(int x,int y)//合并x,y
{
if(!x||!y)return x+y;
if(t[x].val<t[y].val)swap(x,y);//这个根据所须的左偏树是大根堆还是小根堆,这里是小根堆,反过来是大根堆
rs(x)=merge(rs(x),y);//将右子树与y合并
f[rs(x)]=x;//当需要更新父亲时加上
t[x].d=t[rs(x)].d+1;//满足左偏树的性质3
return x;
}
int find(int x) //找到x所在堆的最小值/最大值
{x==f[x]?x:f[x]=find(f[x]);}
void pop(int &x) //删除x节点所在堆的最小值/最大值
{
x=find(x);//找到x所在堆的最大值/最小值
x=merge(t[x].ch[0],t[x].ch[1]);//合并左右儿子
}
ll ans,m;
ll sum[N];
int size[N];
int rt[N];
void dfs(int x)
{
rt[x]=x;
size[x]=1,sum[x]=t[x].val;
for(int i=0;i<v[x].size();i++)
{
if(v[x][i]==x)
continue;
dfs(v[x][i]);
}
for(int i=0;i<v[x].size();i++)
{
if(v[x][i]==x)
continue;
rt[x]=merge(rt[x],rt[v[x][i]]);
sum[x]+=sum[v[x][i]];
size[x]+=size[v[x][i]];
}
while(sum[x]>m)
{
sum[x]-=t[rt[x]].val;
size[x]--;
rt[x]=merge(t[rt[x]].ch[0],t[rt[x]].ch[1]);
}
ans=max(ans,1ll*size[x]*lead[x]);
}
int main()
{
int n;
int a,b,c;
ans=0;
scanf("%d%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&a,&b,&c);
v[a].push_back(i);
t[i].val=b;
lead[i]=c;
}
dfs(1);
cout<<ans<<endl;
}