APIO2012第一题 Dispatching 派遣

描述

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

思路

每个点，他的子树，挑某几个最小的。
对于两个点要合并他们的儿子。
法一 splay。见 http://www.cppblog.com/MatoNo1/archive/2012/05/12/174683.aspx
法二 堆，只取和小于k的点放入堆，和超过了就删最大点。合并时同理。我用的是这个方法，并且用了Priority_Queue。囧。

        一开始我瞎写了Priority_Queue。每个点新建一个。很可能爆空间。

        还是写了一个dfs，很可能爆栈。

        所以写了个数组forq。并改成了bfs。

代码

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define NMAX 100001
long long int M;
int N;
int fa[NMAX];
long long int C[NMAX],L[NMAX];
vector <int > E[NMAX];
priority_queue <pair<int,int > > Q[NMAX];
int forq[NMAX];
long long int sum[NMAX];
long long int ans;
int thor;
int qu[NMAX];
void init()
{
	scanf("%d%I64d",&N,&M);
	ans=0;
	memset(sum,0,sizeof(sum));
	for (int i=0;i<N;i++)
	{
		forq[i]=i;
		E[i].clear();
		//Q[forq[i]].clear();
	}
	for (int i=0;i<N;i++)
	{
		scanf("%d%I64d%I64d",fa+i,C+i,L+i);
		fa[i]--;
		if (fa[i]!=-1) E[fa[i]].push_back(i);
		else thor=i;
	}
}
void solve_one(int a)
{
	int mn=a;
	for (int p=0;p<E[a].size();p++)
		if (mn==a|| Q[forq[mn]].size()<Q[forq[E[a][p]]].size())
			mn=E[a][p];
	forq[a]=forq[mn];sum[a]=sum[mn];

	Q[forq[a]].push(make_pair(C[a],a));
	sum[a]+=C[a];
	while (sum[a]>M) sum[a]-=Q[forq[a]].top().first,Q[forq[a]].pop();

	int t;
	for (int p=0;p<E[a].size();p++)
	{
		t=E[a][p];
		if (t!=mn)
		{
			while (!Q[forq[t]].empty())
			{
				Q[forq[a]].push(Q[forq[t]].top());
				sum[a]+=Q[forq[t]].top().first;
				Q[forq[t]].pop();
				while (sum[a]>M) sum[a]-=Q[forq[a]].top().first,Q[forq[a]].pop();
			}
		}
	}
	ans=max(L[a]*Q[forq[a]].size(),ans);
}
void bfs()
{
	int bot,top,t;
	bot=top=0;
	qu[bot++]=thor;
	while (bot!=top)
	{
		t=qu[top];
		for (int p=0;p<E[t].size();p++)
			qu[bot++]=E[t][p];
		top++;
	}
}
int main()
{
	init();
	bfs();
	for (int i=N-1;i>=0;i--)solve_one(qu[i]);
	printf("%I64d\n",ans);
	return 0;
}