HDU 1520 Anniversary party（简单树形dp）

做刚过类似的但是一直wa，为啥啊，网搜题解多了一个n ！= 0的判断就ac了，可是题干里好像没说啊。。。

思路是利用数组找到根节点，然后dfs跑递归

HDU - 1520

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
using namespace std;
int dp[6005][2];
vector<int> v[6005];
int vis[6005];
void dfs(int cur)
{
    for(int i = 0; i < v[cur].size(); i ++)
    {
        int son = v[cur][i];
        dfs(son);
        dp[cur][1] += dp[son][0];
        dp[cur][0] += max(dp[son][0], dp[son][1]);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n) != EOF && n)
    {
        memset(dp, 0, sizeof(dp));
        memset(vis, 0, sizeof(vis));
        for(int i = 1; i <= n; i ++){
            scanf("%d",&dp[i][1]);
            v[i].clear();
        }
        int u, vv;
        while(scanf("%d%d",&u,&vv) && (u + vv))
        {
            v[vv].push_back(u);
            vis[u] = vv;
        }
        int ans;
        for(int i = 1; i <= n; i ++)
            if(! vis[i]){
                dfs(i);
                ans = i;
                break;
        }
        printf("%d\n",max(dp[ans][0], dp[ans][1]));
    }
    return 0;
}