EOJ 1186 Anniversary party

EOJ 1186 http://acm.cs.ecnu.edu.cn/problem.php?problemid=1186

题意：给定一棵树，选取一定的节点，相邻父子节点不可同时选取。

　　　树型dp，不能简单地用并查集分组。

　　　题解：　　http://blog.csdn.net/woshi250hua/article/details/7641589

//hdu

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <string>
#include <algorithm>
#include <iostream>
#include <map>
#include <queue>

using namespace std;
int f[6005];
struct E
{
    int v;
    int next;
}e[6005];
int head[6005];
int ie;
int dp[6005][2];
int a[6005];

void add(int u, int v)
{
    e[ie].v = v;
    e[ie].next = head[u];
    head[u] = ie++;
    //printf("%d -> %d\n", u, v);
}

void dfs(int t)
{
    //printf("in: %d\n", t);
    //while(getchar() != '.')
        //;
    dp[t][1] = a[t];
    dp[t][0] = 0;
    int p = head[t];
    while(p){
        dfs(e[p].v);
        dp[t][1] += dp[e[p].v][0];
        dp[t][0] += max(dp[e[p].v][0], dp[e[p].v][1]);
        p = e[p].next;
    }
}

int main()
{    
    int n;
    while(cin >> n)
    {
        for(int i=1; i<=n; i++){
            scanf("%d",  &a[i]);
            f[i] = i;
        }
        int u, v;
        ie = 1;
        memset(head, 0, sizeof(head));
        while(scanf("%d%d", &v, &u), v&&u)
        {
            f[v] = u;
            add(u, v);
        }
        int i;
        for(i=1; i<=n; i++)
            if(i == f[i])
            {
                dfs(i);
                break;
            }
        printf("%d\n", max(dp[i][0], dp[i][1]));
    }
    
    return 0;
}