HDU 1520 Anniversary party(树形dp)

题目链接：点击打开链接

题意：由于有上司和下属直接关系的两个人不能同时参加party， 求出party最大的活跃值

每个结点有两种状态，参加和不参加，用0表示不参加，1表示参加

dp[i][1]表示第i个参与者参加了，dp[i][0]表示第i个参与者没有参加。

状态转移方程：dp[u][0] += max (dp[v][0], dp[v][1]) :表示上司没参加，其员工可以参加可以不参加

dp[u][1] += dp[v][0] : 表示若上司参加了，其员工一定不会参加

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
int n,num[6005];
int dp[6005][2];//dp[i][1]:以当前节点i为父节点的 子树，取i时的最大值  dp[i][0]:不取i时的最大值
vector<int> v[6005];
void DP(int x){
	dp[x][1]=num[x];
	dp[x][0]=0;
	for(int i=0;i<v[x].size();i++){
		DP(v[x][i]);
		dp[x][0]+=max(dp[v[x][i]][0],dp[v[x][i]][1]);
		dp[x][1]+=dp[v[x][i]][0];
	}
}
int main(){
	int root,x,y;
	int book[6005];
	while(scanf("%d",&n)!=EOF){
		memset(book,0,sizeof(book));
		for(int i=1;i<=n;i++){
			v[i].clear();
		}
		for(int i=1;i<=n;i++){
			scanf("%d",&num[i]);
		}
		while(scanf("%d%d",&x,&y)!=EOF&&x&&y){
			v[y].push_back(x);
			book[x]=1;
		}
		for(int i=1;i<=n;i++){//找树根
			if(book[i]==0){
				DP(root=i);
				break;
			}
		}
		printf("%d\n",max(dp[root][0],dp[root][1]));
	}
	return 0;
}