Description
在一个忍者的帮派里,一些忍者们被选中派遣给顾客,然后依据自己的工作获取报偿。在这个帮派里,有一名忍者被称之为 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,是可以得到的用户满意度的最大值。
Source
分析:
这道题的题面我参透了很久,还是有些似懂非懂
- 每名忍者都有且仅有一个上级
—> 说明是一个树形结构 - 指令总是由上级发送给他的直接下属;选择一名忍者作为管理者,要求这个管理者可以向所有被派遣的忍者发送指令
—> 说明选择的一定是一棵子树内的结点 - 满意度为派遣的忍者总数乘以管理者的领导力水平
—> 说明如果我们选择了一个结点x作为管理者,那么派遣的队员一定是在ta的子树内 - 还有一个小细节:如果我们不想派遣一个忍者,我们一定是选择要价最高的那个人不工作
我们对题目进行了简单分析之后,就可以得到一个初步的想法:
运用贪心的思想,枚举管理者,
因为我们想要让满意度最大,那么只要能付得起工钱,我们就会选择这个结点
我们不停的选人(注意一定在同一个子树内),并把ta们都加入到一个集合中
一旦付不起工钱,我们就可以不派遣当前集合中要价最高的那个人
很朴素的想法(完全就是题面。。。STO),那我们要怎么实现呢?
- 我们要支持集合与集合之间的合并(因为dfs树在向上回溯的时候需要把儿子节点的信息都合并在一起)
- 我们要支持快速查找集合中的最大元素
- 我们要支持快速删除集合中的最大元素
一个数据结构:左偏树
在得到的树形结构上,我们从底部枚举管理者,
每个结点一开始都是一个集合(即大根左偏树)
同时维护三个数组:
sz:该子树内我们决定派遣的忍者数量
sum:我们需要付给这些忍者的工钱
root:当前结点所在的左偏树的根结点
一开始dfs的时候,我们无脑维护所有的值(就是结点全选的情况)
在回溯的时候,我们判断当前集合是不是可以支付所有忍者的工钱
如果支付不了,我们就果断放弃集合中要价最高的人,让ta只当一个传递者
但是这样的话sum,sz和root都会受影响
然而我们不虚,我们直接删除当前左偏树的根结点
并把ta的贡献从sum和sz中减掉即可
tip
许多左偏树的题隐藏的很深,所以需要认真分析题目,从左偏树的功能入手(插入,删除,合并,查询)
//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
using namespace std;
const int N=100010;
ll pay[N],c[N],ans,sum[N];
int sz[N],root[N],n,m,ch[N][2],dis[N];
struct node{
int x,y,nxt;
};
node way[N<<1];
int st[N],tot=0;
void add(int u,int w)
{
tot++;
way[tot].x=u; way[tot].y=w; way[tot].nxt=st[u]; st[u]=tot;
tot++;
way[tot].x=w; way[tot].y=u; way[tot].nxt=st[w]; st[w]=tot;
}
int merge(int x,int y)
{
if (!x) return y;
if (!y) return x;
if (pay[x]<pay[y]) swap(x,y); //大根堆
ch[x][1]=merge(ch[x][1],y);
if (dis[ch[x][1]]>dis[ch[x][0]]) swap(ch[x][0],ch[x][1]);
dis[x]=dis[ch[x][1]]+1;
return x;
}
void dfs(int now,int fa)
{
sz[now]=1; //该子树内我们决定派遣的忍者数量
root[now]=now; //我们需要付给这些忍者的工钱
sum[now]=pay[now]; //当前结点所在的左偏树的根结点
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa)
{
dfs(way[i].y,now);
sz[now]+=sz[way[i].y];
sum[now]+=sum[way[i].y];
root[now]=merge(root[now],root[way[i].y]);
}
while (sum[now]>m)
{
sum[now]-=pay[root[now]];
sz[now]--;
root[now]=merge(ch[root[now]][0],ch[root[now]][1]);
}
ans=max(ans,sz[now]*c[now]);
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
{
int x,y,z;
scanf("%d%lld%lld",&x,&pay[i],&c[i]);
if (x) add(i,x);
}
dfs(1,0);
printf("%lld",ans);
return 0;
}