Time Limit: 10 Sec
Memory Limit: 128 MB
Description
在一个忍者的帮派里,一些忍者们被选中派遣给顾客,然后依据自己的工作获取报偿。
在这个帮派里,有一名忍者被称之为Master。除了Master以外,每名忍者都有且仅有一个上级。为保密,同时增强忍者们的领导力,所有与他们工作相关的指令总是由上级发送给他的直接下属,而不允许通过其他的方式发送。
现在你要招募一批忍者,并把它们派遣给顾客。你需要为每个被派遣的忍者支付一定的薪水,同时使得支付的薪水总额不超过你的预算。另外,为了发送指令,你需要选择一名忍者作为管理者,要求这个管理者可以向所有被派遣的忍者发送指令,在发送指令时,任何忍者(不管是否被派遣)都可以作为消息的传递人。管理者自己可以被派遣,也可以不被派遣。当然,如果管理者没有被排遣,你就不需要支付管理者的薪水。
你的目标是在预算内使顾客的满意度最大。这里定义顾客的满意度为派遣的忍者总数乘以管理者的领导力水平,其中每个忍者的领导力水平也是一定的。
写一个程序,给定每一个忍者i的上级Bi,薪水Ci,领导力Li,以及支付给忍者们的薪水总预算M,输出在预算内满足上述要求时顾客满意度的最大值。
Input
第一行包含两个整数N和M,其中N表示忍者的个数,M表示薪水的总预算。
接下来N行描述忍者们的上级、薪水以及领导力。其中的第i行包含三个整数Bi,Ci,Li分别表示第i个忍者的上级,薪水以及领导力。Master满足Bi=0,并且每一个忍者的老板的编号一定小于自己的编号Bi
Output
输出一个数,表示在预算内顾客的满意度的最大值。
HINT
1 ≤ N ≤ 100,000 忍者的个数;
1 ≤ M ≤ 1,000,000,000 薪水总预算;
0 ≤ Bi < i 忍者的上级的编号;
1 ≤ Ci ≤ M 忍者的薪水;
1 ≤ Li ≤ 1,000,000,000 忍者的领导力水平。
对于 30%的数据,N ≤ 3000。
题目分析
题目大意就是选择一批忍者,以及他们的一个公共祖先(管理者)
使得总费用不超过m且忍者个数与公共祖先领导力乘积最大
可以先确定大致的思路为
枚举每个点为管理者,并在他的子树内选择尽量多的节点
显然要使节点尽量多就要先选费用少的
但直接在子树内寻找会很麻烦
所以把思路反过来
先假设以u为根的子树全部选
然后判断总费用是否大于m
若是则一直弹出费用最大的
直到费用小于等于m
然后更新答案
ans=min(ans,li[u]∗sz[u])
a
n
s
=
m
i
n
(
a
n
s
,
l
i
[
u
]
∗
s
z
[
u
]
)
再把以u为根选出的节点继续向上合并
要同时支持合并和最大值查询,不难想到左偏树
所以这题就是dfs到根,再用左偏树一直向上合并
在这个过程中不断更新答案
由于题目中保证
Bi<i
B
i
<
i
所以可以用手工栈简单的模拟递归
直接循环,节点顺序从
n−1
n
−
1
就可以
这样也能避免递归导致的爆栈
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
lt read()
{
lt f=1,x=0;
char ss=getchar();
while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
return f*x;
}
const int maxn=200010;
int n;
struct node{int v,nxt;}E[maxn<<1];
int head[maxn],tot;
lt ci[maxn],li[maxn],m;
int rt[maxn],cnt;
int val[maxn],dis[maxn],ch[maxn][2];
lt sz[maxn],sum[maxn];
lt ans;
void add(int u,int v)
{
E[++tot].nxt=head[u];
E[tot].v=v;
head[u]=tot;
}
int merge(int x,int y)
{
if(!x||!y)return x+y;
if(val[x]<val[y])swap(x,y);
ch[x][1]=merge(ch[x][1],y);
if(dis[ch[x][0]]<dis[ch[x][1]]) swap(ch[x][0],ch[x][1]);
dis[x]=dis[ch[x][1]]+1;
return x;
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;++i)
{
int ff=read();
ci[i]=read();li[i]=read();
add(ff,i);
}
for(int u=n;u>=1;--u)
{
rt[u]=++cnt; val[cnt]=sum[u]=ci[u]; sz[u]=1;
for(int i=head[u];i;i=E[i].nxt)
{
int v=E[i].v;
sum[u]+=sum[v]; sz[u]+=sz[v];
rt[u]=merge(rt[u],rt[v]);
}
while(sum[u]>m)
{
sum[u]-=val[rt[u]]; --sz[u];
rt[u]=merge(ch[rt[u]][0],ch[rt[u]][1]);
}
ans=max(ans,sz[u]*li[u]);
}
printf("%lld",ans);
return 0;
}