[BZOJ 2809]dispatching(启发式合并)

Description

在一个忍者的帮派里，一些忍者们被选中派遣给顾客，然后依据自己的工作获取报偿。在这个帮派里，有一名忍者被称之为 Master。除了 Master以外，每名忍者都有且仅有一个上级。为保密，同时增强忍者们的领导力，所有与他们工作相关的指令总是由上级发送给他的直接下属，而不允许通过其他的方式发送。现在你要招募一批忍者，并把它们派遣给顾客。你需要为每个被派遣的忍者 支付一定的薪水，同时使得支付的薪水总额不超过你的预算。另外，为了发送指令，你需要选择一名忍者作为管理者，要求这个管理者可以向所有被派遣的忍者 发送指令，在发送指令时，任何忍者（不管是否被派遣）都可以作为消息的传递 人。管理者自己可以被派遣，也可以不被派遣。当然，如果管理者没有被排遣，就不需要支付管理者的薪水。你的目标是在预算内使顾客的满意度最大。这里定义顾客的满意度为派遣的忍者总数乘以管理者的领导力水平，其中每个忍者的领导力水平也是一定的。写一个程序，给定每一个忍者 i的上级 Bi，薪水Ci，领导力L i，以及支付给忍者们的薪水总预算 M，输出在预算内满足上述要求时顾客满意度的最大值。

1 ≤N ≤ 100,000 忍者的个数；
1 ≤M ≤ 1,000,000,000 薪水总预算；
0 ≤Bi < i 忍者的上级的编号；
1 ≤Ci ≤ M 忍者的薪水；
1 ≤Li ≤ 1,000,000,000 忍者的领导力水平。

Input
从标准输入读入数据。

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

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

Output
输出一个数，表示在预算内顾客的满意度的最大值。

Sample Input

5  4

0  3 3

1  3 5

2  2 2

1  2 4

2  3 1

Sample Output

6 

思路
训练用板题。
可以将题目转化为给定一棵规模达10^5个节点的树，每个节点有两个参数(x,y)。
要求选择一颗子树(以u为根节点）中的点集S，使得在满足∑xi<X （xi∈S）的前提下，u.y*size(S)最大。
对于每个节点开一个优先队列（或set替代）存储所有子树中的节点，并记录队列元素的总和S。
当S大于限制时，弹出元素直至剩余元素满足要求。
（启发式合并的精髓）在合并所有子树的节点集合时，将小集合合并至大集合以降低时间开销
很奇怪的一点是：不使用启发式合并的话会MLE而不是TLE，占用的内存估计是10倍以上。
考虑到用结构体可能会出现一些奇奇怪怪的问题，这里用对象来传递封装的参数，用指针来避免因为复制对象而可能出现的内存爆炸。
但事实上，不用指针的版本反而跑得更慢，内存更小

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef struct data{
	int f,m,bro,son,_son,le;
}data;
data d[100005];
int n,m;
class s{
	public:
	priority_queue<int>q;
	long long sum;
	s()
	{
		sum=0;while(!q.empty())q.pop();
	}
};
long long ans;
s *dfs(int now)
{
	s *temp=new s;
	temp->q.push(d[now].m);
	temp->sum+=(d[now].m);
	for(int i=d[now].son;i>0;i=d[i].bro)
	{
		s *temp2=dfs(i);
		if(temp->q.size()<temp2->q.size())swap(temp,temp2);
		temp->sum+=temp2->sum;
		while(!temp2->q.empty())
		{
			temp->q.push(temp2->q.top());
			temp2->q.pop();
		}
	}
	while(temp->sum>m)
	{
		temp->sum-=temp->q.top();
		temp->q.pop();
	}
	ans=max(ans,(long long)(temp->q.size())*d[now].le);
	return temp;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) d[i].f=d[i].bro=d[i].son=d[i]._son=-1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&d[i].f,&d[i].m,&d[i].le);
		if(d[i].f)
		{
			if(d[d[i].f].son!=-1)
			{
				d[d[d[i].f]._son].bro=i;
				d[d[i].f]._son=i;
			}else
			{
				d[d[i].f]._son=d[d[i].f].son=i;
			}
		}
	}
	dfs(1);
	printf("%lld",ans);
	return 0;
}