HDU1520(树形dp入门）

解题思路：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;
}