hdu1520 Anniversary party

Anniversary party

HDU - 1520

题意：你要举行一个晚会，所有人的关系可以构成一棵树，要求上下级关系的人不能同时出现，每一个人都有一个rating值，要求使整个晚会的rating值最大。

/*
    解题思路：树形dp，随意从一个点开始扩展，把周围所有节点的dp都解决出来，然后加上去即可。
    用dp[i][0]表示不选中i号，周围的都可以选的最大值，dp[i][1]表示选中i号，那么周围都不能选的最大值。
    则
    dp[i][0]+=(dp[j][1],dp[j][0])
    dp[i][1]+=dp[j][0]
    i和j相邻，并且在求i的时候从j扩展的都已经求出来。
*/
#include<cstdio>
#include<iostream>
#include<vector>
#include<string.h>
using namespace std;
#define Max(a,b) a>b?a:b
vector<int> q[];
int visit[],rating[],dp[][];
void dfs(int r)
{
    //printf("r=%d\n",r);
    int i,v;
    visit[r]=;
    if(q[r].size()==)
    {
        dp[r][]=rating[r];
        dp[r][]=;
    }
    int dp1=,dp0=;
    for(i=;i<q[r].size();i++)
    {
        v=q[r][i];
        if(!visit[v])
        {
            //printf("v=%d ",v);
            dfs(v);
        }
        dp1+=dp[v][];//若选了此节点，则孩子都不能选
        dp0+=Max(dp[v][],dp[v][]);//若不选此节点，则孩子可选可不选
    }
    dp[r][]=dp1+rating[r];
    dp[r][]=dp0;
}
int main()
{
    int n,root;
    int i,j,k;
    while(~scanf("%d",&n))
    {
        for(i=;i<=n;i++)
        q[i].clear();
        for(i=;i<=n;i++)
        scanf("%d",&rating[i]);
        memset(visit,,sizeof(visit));
        int c,f;
        root=;
        while(scanf("%d%d",&c,&f),c||f)
        {
            if(c==root)
            root=f;//新学习的一个寻根的小技巧
            q[f].push_back(c);
        }
        dfs(root);
        printf("%d\n",Max(dp[root][],dp[root][]));
    }
    return ;
}