[HDU]1520 Anniversary party

這道題目其實就是一個裸的樹形DP，和洛谷P1352 沒有上司的舞會一模一樣，只是要加一個讀入判斷罷了[英文不好導致錯了3回，每次都是Wrong Answer，就是沒有加這個判斷！！！]

#include<bits/stdc++.h>
using namespace std;
int dp[6010][2],fa[6010],v[6010],n;
void tree_dp(int i){
    int j;
    for(j=1;j<=n;j++)
        if(fa[j]==i){
            tree_dp(j);
            dp[i][1]+=dp[j][0];
            dp[i][0]+=max(dp[j][0],dp[j][1]);
        }
}
int main(){
    int i,j,k,m;
    while(scanf("%d",&n)==1){
        memset(fa,-1,sizeof(fa));
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++)
            scanf("%d",&dp[i][1]);
        int gen=0;
        int l;
        while(scanf("%d%d",&l,&k)==2 && l && k){
            fa[l]=k;
            if(!gen || fa[gen]==k)
                gen=k;
        }
        while(fa[gen]!=-1)gen=fa[gen];
        tree_dp(gen);
        printf("%d\n",max(dp[gen][0],dp[gen][1]));
    }
    return 0;
}