P1552 [APIO2012]派遣 题解

题目

左偏树 + 树形DP

这题的思路还是比较好想的
首先，对于每一个节点维护一个左偏树（大根堆），同时维护堆内忍者个数和费用和
接着，我们用 dfs 从下往上合并
当我们合并到点 $x$ 时，有可能目前堆内的总费用之和超出预算，所以我们利用贪心的思想将费用最高的忍者依次弹出，直到总费用不超过预算

设 $s[x],c[x],id[x]$ 分别为 $x$ 所在的堆中忍者数，费用和，和该堆的根节点

关键代码

void dfs(long long x)
{
	s[x]=1,c[x]=a[x]; // a[x] 为忍者x所需的费用
	for(long long i=0;i<e[x].size();++i)
	{
		long long y=e[x][i];
		dfs(y);
		id[x]=merge(id[x],id[y]); //合并所有子节点所在的堆
		s[x]+=s[y],c[x]+=c[y];// 更新
	}
	while(c[x]>m) // 将最贵的忍者依次弹出，直到不超过预算位置
	{
		--s[x],c[x]-=v(id[x]);
		id[x]=del(id[x]);
	}
	ans=max(ans,b[x]*s[x]);
}

注：这里的 费用和与堆中人数 都是在左偏树外面维护的，个人认为比较简洁，其实也可以在堆内维护

完整代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const long long Maxn=200000+10,inf=0x3f3f3f3f;
struct node{
	long long ls,rs,val,fa,dis;
	#define ls(x) g[x].ls
	#define rs(x) g[x].rs
	#define v(x) g[x].val
	#define f(x) g[x].fa
	#define dis(x) g[x].dis
}g[Maxn];
long long a[Maxn],b[Maxn];
long long s[Maxn],c[Maxn],id[Maxn];
vector <long long> e[Maxn];
long long n,m,root,ans;
inline long long read()
{
	long long s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
void init()
{
	for(long long i=1;i<=n;++i)
	f(i)=i,v(i)=a[i];
}
long long merge(long long x,long long y)
{
	if(!x || !y)return x|y;
	if(v(x)<v(y))swap(x,y);
	rs(x)=merge(rs(x),y);
	if(dis(rs(x))>dis(ls(x)))swap(ls(x),rs(x));
	if(!rs(x) || !ls(x))dis(x)=0;
	else dis(x)=dis(rs(x))+1;
	return x;
}
long long del(long long x)
{
	long long l=ls(x),r=rs(x);
	f(l)=l,f(r)=r;
	ls(x)=rs(x)=dis(x)=0;
	return merge(l,r);
}
void dfs(long long x)
{
	s[x]=1,c[x]=a[x];
	for(long long i=0;i<e[x].size();++i)
	{
		long long y=e[x][i];
		dfs(y);
		id[x]=merge(id[x],id[y]);
		s[x]+=s[y],c[x]+=c[y];
	}
	while(c[x]>m)
	{
		--s[x],c[x]-=v(id[x]);
		id[x]=del(id[x]);
	}
	ans=max(ans,b[x]*s[x]);
}
int main()
{
//	freopen("in.txt","r",stdin);
//	freopen("out.txt","w",stdout);
	n=read(),m=read();
	dis(0)=-1;
	for(long long i=1;i<=n;++i)
	{
		long long x=read();
		a[i]=read(),b[i]=read();
		if(x)e[x].push_back(i);
		else root=i;
		id[i]=i;
	}
	init();
	dfs(root);
	printf("%lld\n",ans);
	return 0;
}