解题思路:https://blog.csdn.net/yexiaohhjk/article/details/52694214
设dp[i][0]表示:当前这个点不选,dp[i][1]表示当前这个点选择的最优解。
转移方程:dp[cur][0]+=max(dp[son][1],dp[son][0]);//当前这个点不选,那他的孩子可选可不选,取
最大的。
dp[cur][1]+=dp[son][0]//当前这点选择,那他的孩子就不能选择。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> s[6005];
int q[6005],fa[6005];
int dp[6005][2];
void dfs(int r)
{
dp[r][1]=q[r];
for(int i=0;i<s[r].size();i++)
{
int son=s[r][i];
dfs(son);
dp[r][0]+=max(dp[son][0],dp[son][1]);
dp[r][1]+=dp[son][0];
}
}
int main()
{
//freopen("t.txt","r",stdin);
int n,a,b;
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d",&q[i]);
dp[i][0]=dp[i][1]=0;
s[i].clear();
fa[i]=-1;
}
while(1)
{
scanf("%d%d",&a,&b);
if(a+b==0) break;
s[b].push_back(a);
fa[a]=b;
}
int t=1;
while(fa[t]!=-1) t=fa[t];
dfs(t);
printf("%d\n",max(dp[t][1],dp[t][0]));
}
return 0;
}