poj 2342 Anniversary party_经典树状dp

题意：Ural大学有n个职员，1~N编号，他们有从属关系，就是说他们关系就像一棵树，父节点就是子节点的直接上司，每个职员有一个快乐指数，现在要开会，职员和职员的直接上司不能同时开会，问怎才能使开会的快乐指数最高。

思路：用树状dp，dp[i][0]为不参加会议，dp[i][1]为参加会议

#include<iostream>
#include<cstdio>
#include<vector>
#define N 6010
using namespace std;
vector<int> v[N];
int dp[N][2],hp[N],mark[N];
int max(int a,int b){return a>b?a:b;}
void dfs(int k)
{
	int i,len=v[k].size();
	dp[k][0]=0;
	dp[k][1]=hp[k];
	if(len==0)
		return ;
	for(i=0;i<len;i++)
	{
		dfs(v[k][i]);
		dp[k][0]+=max(dp[v[k][i]][1],dp[v[k][i]][0]);//不参加会议，下属有可以参加会议，也可以不参加会议
		dp[k][1]+=dp[v[k][i]][0];//参加会议，下属不能参加会议

	}
}
int main()
{
	int n,a,b,i,k;
	while(scanf("%d",&n)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%d",&k);
			hp[i]=k;
			v[i].clear();
		}
		memset(mark,0,sizeof(mark));
		while(scanf("%d%d",&b,&a)&&(a||b))
		{
			v[a].push_back(b);
			mark[b]=1;
		}
		for(i=1;i<=n;i++)//找出根节点
			if(!mark[i])
				break;
		dfs(i);
		printf("%d\n",max(dp[i][0],dp[i][1]));
	}
	return 0;
}