poj 2342 Anniversary party (树形dp入门)

http://poj.org/problem?id=2342

树形dp:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

const int mx=6006;
int dp[mx][2];
vector<int>g[mx];
int n;

void dfs(int x)
{
   for (int i=0;i<g[x].size();i++)
   {
       int j=g[x][i];
       dfs(j);
       dp[x][1]+=dp[j][0];
       dp[x][0]+=max(dp[j][1],dp[j][0]);
   }
}

int main()
{
    while (~scanf("%d",&n))
    {
        if (!n)
        {
            scanf("%d",&n);
            if (!n) return 0;
        }
        memset(dp,0,sizeof(dp));
        for (int i=1;i<=n;i++) scanf("%d",&dp[i][1]),g[i].clear();
        int l,k,root=1;
        for (int i=1;i<n;i++)
        {
            scanf("%d%d",&l,&k);
            g[k].push_back(l);
            if (root==l) root=k;
        }
        dfs(root);
        printf("%d
",max(dp[root][1],dp[root][0]));
    }
    return 0;
}