hdu1520 Anniversary party (树形dp)

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=1520
题意：上司和直系下属不能同时参加party，求party的最大活跃值。
输入： 输入n个结点，接下来的n行，表示1~n的每个结点分别具有的活跃值，输入a,b，表示b是a的上司，当a==0&& b==0时，该组数据输入结束。
输出：由于直接有上司和下属关系的两个人不能同时参加party， 求出能让party活跃值最大的方案（求出最大的活跃值即可）.
状态转移方程： dp[i][1]+=dp[j][0]； 　　　　　　　 

　　　　　　　 dp[i][0]+=max(dp[j][0],dp[j][0]);

其中i是j的父亲结点。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=;

struct node
{
    int v;
    int next;
} node[N<<];

int dp[N][];
int a[N],head[N],in[N];
int k;

void addedge(int u,int v)  ///u是父亲，v是孩子
{
    node[k].v=v;
    node[k].next=head[u];
    head[u]=k++;
}

int dfs(int u)
{
    dp[u][]=;
    dp[u][]=a[u];
    for(int i=head[u]; i!=-; i=node[i].next)
    {
        int v=node[i].v;
        dfs(v);
        dp[u][]+=max(dp[v][],dp[v][]);
        dp[u][]+=dp[v][];
    }
    return max(dp[u][],dp[u][]);
}

int main()
{
    int n,x,y;
    while(scanf("%d",&n)==)
    {
        k=;
        memset(head,-,sizeof(head));
        memset(dp,,sizeof(dp));
        memset(in,,sizeof(in));
        for(int i=; i<=n; i++)
            scanf("%d",&a[i]);
        while(scanf("%d%d",&x,&y) && (x+y))
        {
            addedge(y,x);
            in[x]++;
        }
        int ans=;
        for(int i=; i<=n; i++)  ///其实这是一个森林，找到每棵树的根，求和
            if(!in[i]) ans+=dfs(i);
        printf("%d\n",ans);
    }
    return ;
}