原题链接:https://vjudge.net/problem/19758/origin
树形DP:
#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 6050;
vector<int>vec[maxn];
int f[maxn];
int hap[maxn];
int dp[maxn][2];
void dfs(int root)
{
int len = vec[root].size();
dp[root][1]=hap[root];
for(int i=0;i<len;i++)
{
dfs(vec[root][i]);
}
for(int i=0;i<len;i++)
{
dp[root][0]+=max(dp[vec[root][i]][1],dp[vec[root][i]][0]);
dp[root][1]+=dp[vec[root][i]][0];
}
}
int main()
{
int n;
int a,b;
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d",&hap[i]);
vec[i].clear();
f[i]=-1;//树根标记
dp[i][0]=dp[i][1]=0;
}
while(scanf("%d%d",&a,&b))
{
if(a==0&&b==0) break;
f[a]=b;
vec[b].push_back(a);
}
a=1;
while(f[a]!=-1) a=f[a];//找到树根
dfs(a);
cout<<max(dp[a][1],dp[a][0])<<endl;
}
return 0;
}