POJ 2342 URAL 1039 Anniversary Party

    链接：POJ - 2342     URAL 1039

    题意：在一个公司里面，每个人都带有一定的快乐值，现在举行一个宴会，为了使宴会的总快乐值最大，不让宴会里的每个人的直接上司参加宴会，问宴会能达到的最大快乐值是多少。

    输入数据中：首先给一个数 N，然后有 N 行，代表 第 1 到 第 N 个人的快乐值，接下来每行有两个数 a ,b ，代表 b 是 a 的直接上司，最后一行以 0 0 结束。

    思路：整个公司的人，形成了一棵树，每个节点带有一个权值，选取的方法要求父亲和孩子不能同时出现。可以想到，这是树形 Dp，我们可以用  dp[i][0] 表示不选 i 节点所能获得的最大值，dp[i][1] 表示选 i 节点获得的最大值。假设 k 是 i 的孩子，如果取了 i， k 肯定不能取， dp[i][1] += dp[k][0]，如果不取 i ，k 可能取也可能不取，dp[i][0] += max（dp[i][1],dp[i][0])。注意 URAL 卡内存，而且测试数据更难过哦。struct 不能用 vector，怒用 class...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>

using namespace std;

#define maxn 6003

class node
{
public:
	int num,rate,fa,k;
	vector<int> child;
}person[maxn];

int n,real_root,root[maxn],dp[maxn][2];

void solve(int r)
{
	dp[r][0] = 0;
	dp[r][1] = person[r].rate;
	for(int i = 0;i < person[r].k;++i)
	{
		int child = person[r].child[i];
		solve(child);
		dp[r][0] += max(dp[child][0],dp[child][1]);
		dp[r][1] += dp[child][0];
	}
}


int main()
{
	while(~scanf("%d",&n))
	{
		memset(dp,0,sizeof(dp));
		for(int i = 1;i <= n;++i)
		{
			scanf("%d",&person[i].rate);
			person[i].num = i;
			person[i].k = 0;
			root[i] = 1;
		}
		int t1,t2;
		while(scanf("%d%d",&t1,&t2) && t1)
		{
			person[t1].fa = t2;
			person[t2].child.push_back(t1);
			root[t1] = 0;
			person[t2].k++;
		}
            //找根节点
		for(int i = 1;i <= n;++i)
		{
			if(root[i])
			{
				real_root = i;
				break;
			}
		}
		solve(real_root);
		cout<<max(dp[real_root][0],dp[real_root][1])<<endl;
	}
	return 0;

}