POJ 2342 Anniversary party ~~Tree Dp

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/cnyali/article/details/77816016
/*
【题意】
公司有n个人,每个人有价值vi,有一天举办年会,每个人都可以参加,但有严格的等级制度,参加活动时,不能同时出现a和a的上司,问如何才能使总和最大。
【分析】
每个人只有去和不去两种状态,设f[i][0]和f[i][1]分别表示第i个人不参加和参加年会,获得的总的最大价值。
则状态转移方程为:
f[i][1] += f[j][0],
f[i][0] += max{f[j][0],f[j][1]};
其中j为i的孩子节点。
这样,从根节点root进行dfs,最后结果为max{f[root][0],f[root][1]}。
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
const int maxn=6000+5;
using namespace std;
struct node{
	int fa;
	vector<int>ch;
}a[maxn];
int f[maxn][2],p[maxn];
void dfs(int x){
	p[x]=1;
	int s=a[x].ch.size();
	for(int i=0;i<s;i++){
		int k=a[x].ch[i];
		if(!p[k]){
			dfs(k);
			f[x][1]+=f[k][0];
			f[x][0]+=max(f[k][0],f[k][1]);
		}
	}
}
int main(){
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
	int i,j,k,m,n,x,y,root=1;
	cin>>n;
	for(i=1;i<=n;i++)scanf("%d",&f[i][1]);	
	while(scanf("%d%d",&x,&y),x+y>0){
		a[x].fa=y;	a[y].ch.push_back(x);
	}
	for(i=1;i<=n;i++)
		if(a[i].fa==0){root=i;break;}
	dfs(root);
	printf("%d\n",max(f[root][0],f[root][1]));
	return 0;
}

猜你喜欢

转载自blog.csdn.net/cnyali/article/details/77816016