Jzoj5426 摘Galo

0v0在野外看到了一棵Galo树,看到食物的0v0瞪大了眼睛,变成了OvO。
这棵Galo树可以看做是一棵以1号点为根的n个点的有根数,除了根节点以外,每个节点i都有一个Galo,美味度为w[i]。
OvO发现,如果她摘下了i号Galo,那么i的子树中的Galo以及i到根的路径上的其他Galo都会死掉。
OvO的袋子只能装k个Galo,她的嘴巴里还能叼1个,请问她所摘Galo的美味度之和的最大值是多少?

基本树形背包,设f[i][j]表示以i为根的子树,选了j个galo的最大美味度

那么,f[i][j]=max(f[i][k],f[v][j-k]),f[i][1]=max(f[i][1],w[i])

注意两个优化

1.题目中的n,k限制比较诡异:n*k<=10^7

于是这道题f不能直接开要开成一维的,访问f[i][j]要变成f[ik+j]

2.树形背包直接做是会T的,要么跑出dfs序变成序列做法,或者是减少背包状态

我们令size'[x]表示以x为根的叶子结点个数

那么显然多过size'[x]的部分是可以不用的,这样就可以防止别人用几条链的数据来卡你

#include<vector>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 100010
#define f(i,j) F[inv[i]+j-1]
using namespace std;
vector<int> s[N];
int n,_k,val[N]={0},sz[N],inv[N];
long long F[N*200];
int dfs(int x){
	sz[x]=0;
	if(s[x].size()==0){
		f(x,0)=0;
		f(x,1)=val[x];
		return ++sz[x];
	}
	for(int v,i=0,z=s[x].size();i<z;++i){
		sz[x]+=dfs(v=s[x][i]);
		for(int j=min(_k,sz[x]);j;--j)
			for(int k=min(j,sz[v]);k;--k)
				f(x,j)=max(f(x,j),f(x,j-k)+f(v,k));
	}
	f(x,1)=max(f(x,1),(long long)val[x]);
	return sz[x];
}
int main(){
	freopen("galo.in","r",stdin);
	freopen("galo.out","w",stdout);
	scanf("%d%d",&n,&_k); ++_k; _k=min(n,_k);
	for(int i=1;i<=n;++i) inv[i]=(_k+1)*i;
	for(int x,y,i=2;i<=n;++i){
		scanf("%d%d",&x,val+i);
		s[x].push_back(i); 
	}
	dfs(1);
	for(int i=1;i<=_k;++i) F[0]=max(F[0],f(1,i));
	printf("%lld\n",*F);
}

posted @ 2017-11-08 16:15  扩展的灰(Extended_Ash)  阅读(198)  评论(0编辑  收藏  举报