HDU 1520 Anniversary party

题意：n个人去参加聚会，其中有直接上下级关系的参加聚会会影响气氛，所以不能参加，每个人参加聚会都会有一个活跃度，现在要你计算从n个人中选择出一些人，使得本次聚会的活跃度最大

输入数据：
第一行一个N代表有N个人，然后是N个人每个人的活跃度。
然后再N-1行， 每行两个数字，L K 代表 第K个员工是第L个人的上司。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long LL;
const int INF = 1e9+7;
const int maxn = 6010;
const int MOD = 1e9+7;
int n, dp[2][maxn], In[maxn];///dp[这个点是否选取][第n个点]
vector<vector<int> > G;

void TreeDp(int root)
{
int len = G[root].size();
for(int i=0; i<len; i++)
{
int v = G[root][i];
TreeDp(v);
dp[0][root] += max(dp[0][v],dp[1][v]);
dp[1][root] += dp[0][v];
}
}

int main()
{
int a, b;
while(scanf("%d", &n) != EOF)
{
G.clear();
G.resize(n+5);
memset(dp, 0, sizeof(dp));
memset(In, 0, sizeof(In));
for(int i=1; i<=n; i++)
scanf("%d", &dp[1][i]);

while(scanf("%d %d", &a, &b), a+b)
{
G[b].push_back(a);
In[a] ++;
}
int root;

for(int i=1; i<=n; i++)
if(!In[i]) root = i;

TreeDp(root);

printf("%d\n", max(dp[0][root],dp[1][root]));

}
return 0;
}