HDU-1520 Anniversary party（树dp）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1520
题意：开party……。多个下属对应其上司，邀请了上司参加，其下属就不能参加，每个人有个value，求最大的value和
思路：树dp
dp[u][0]没有被邀请的最大值，dp[u][1]被邀请的最大值
dp[u][0] += max(dp[v][0],dp[v][1]);//父亲没有被邀请
dp[u][1] += dp[v][0];//父亲被邀请了，儿子不能被邀请
代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N = 1e5+5;

vector<int> G[N];
int dp[N][2],p[N],f[N];
void dfs(int u,int fa)
{
    dp[u][0] = 0;
    dp[u][1] = p[u];
    for(auto v : G[u])
    {
        if(v == fa)
            continue;
        dfs(v,u);
        dp[u][0] += max(dp[v][0],dp[v][1]);//父亲没有被邀请
        dp[u][1] += dp[v][0];//父亲被邀请了，儿子不能被邀请
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i = 1;i <= n;i++)
            G[i].clear(),f[i] = -1;
        for(int i = 1;i <= n;i++)
            scanf("%d",&p[i]);
        int u,v;
        while(~scanf("%d%d",&u,&v) && (u||v))
        {
            f[u] = v;
            G[v].push_back(u);
        }
        int x = 1;//找根结点
        while(f[x] != -1)
            x = f[x];
        dfs(x,-1);
        printf("%d\n",max(dp[x][0],dp[x][1]));
    }
    return 0;
}