poj(2342)——Anniversary party

题意：

现在有n个人想要参加一个聚会，而且每个人的关系组成了一个树形图，并且每个人想要不和自己的直接上司待在一起，每个人有一个高兴值，问如何分配使得总的高兴值最大，并输出这个最大的高兴值。

思路：

dp[i][1]:代表的是选择第i个人，因为我们尽可能的想让直接上司不在，所以我们不选择i的儿子，所以加上的是dp[son][0]；

dp[i][0]:代表的是不选择第i个人，这个时候我们可以选择第i个人的儿子，所以dp[i][0]+=max(dp[son][1],dp[son][0])；

然后利用并查集，首先找出这n个人的boss，然后再用dfs搜索的方式，然后进行递归，最后输出的是max(dp[root][1],dp[root][0]);

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
#define maxn 6010
int dp[maxn][2],par[maxn];
int vis[maxn];
int ans=-1,n;
void init(){
	for(int i=0;i<=maxn;i++) par[i]=i;
}
int find(int x){
	return x==par[x]?x:find(par[x]);
}
void creat(int root){
	vis[root]=1;
	for(int i=1;i<=n;i++){
		if(!vis[i]&&par[i]==root){
			creat(i);
			dp[root][1]+=dp[i][0];
			dp[root][0]+=max(dp[i][1],dp[i][0]);
		}
	}
}
int main(){
	while(~scanf("%d",&n)){
		memset(dp,0,sizeof(dp));
		memset(vis,0,sizeof(vis));
		init();
		ans=-1;
		for(int i=1;i<=n;i++){
			int x;
			scanf("%d",&x);
			dp[i][1]=x;
		}
		int l,k;
		int root=0;
		while(~scanf("%d%d",&l,&k)){
			if(l==0&&k==0) break;
			par[l]=k;
		}
		root=find(n);
		creat(root);
		int ans=max(dp[root][1],dp[root][0]);
		printf("%d\n",ans);
	}
}