poj 2342 Anniversary party

题目链接：http://poj.org/problem?id=2342

题意：读题很容易懂，这里不做介绍。

解法：树形DP之路的第一道题。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define inf 0x7fffffff
using namespace std;
const int maxn=6010;
int n;
int f[maxn][2],vis[maxn];
int father[maxn];
void dfs(int root)
{
    vis[root]=1;
    for (int i=1 ;i<=n ;i++)
    {
        if (!vis[i] && father[i]==root)
        {
            dfs(i);
            f[root][0] += max(f[i][1],f[i][0]);
            f[root][1] += f[i][0];
        }
    }
}
int main()
{
    while (cin>>n)
    {
        memset(f,0,sizeof(f));
        memset(vis,0,sizeof(vis));
        for (int i=1 ;i<=n ;i++) father[i]=i;
        for (int i=1 ;i<=n ;i++) cin>>f[i][1];
        int a,b;
        int root;
        while (cin>>a>>b)
        {
            if (!a && !b) break;
            father[a]=b;
        }
        for (int i=1 ;i<=n ;i++) if (father[i]==i) root=i;
        dfs(root);
        cout<<max(f[root][0],f[root][1])<<endl;
    }
    return 0;
}