这道题目其实就是一个裸的树形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;
}