动态规划（树型DP，week13作业选做）

题目

在大家的三连助攻下，TT 一举获得了超级多的猫咪，因此决定开一间猫咖，将快乐与大家一同分享。并且在开业的那一天，为了纪念这个日子，TT 在猫咖门口种了一棵苹果树。

一年后，苹果熟了，到了该摘苹果的日子了。

已知树上共有 N 个节点，每个节点对应一个快乐值为 w[i] 的苹果，为了可持续发展，TT 要求摘了某个苹果后，不能摘它父节点处的苹果。

TT 想要令快乐值总和尽可能地大，你们能帮帮他吗？

结点按 1～N 编号。

第一行为 N (1 ≤ N ≤ 6000) ，代表结点个数。

接下来 N 行分别代表每个结点上苹果的快乐值 w[i]（-128 ≤ w[i] ≤ 127)。

接下来 N-1 行，每行两个数 L K，代表 K 是 L 的一个父节点。

输入有多组，以 0 0 结束。

每组数据输出一个整数，代表所选苹果快乐值总和的最大值。

7
1
1
1
1
1
1
1
1 3
7 4
2 3
4 5
6 4
3 5
0 0



5

思路

这题和上司舞会问题相似，使用树型DP解决。
注意：记得初始化

代码

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;


const int maxn=7000;
int w[maxn];
vector<int> v[maxn];//存储每个节点的孩子节点 
int dp[maxn][2];
int root[maxn];
int mark[maxn];

void init()
{
	for(int i=0;i<maxn;i++)
	{
		w[i]=0;v[i].clear();
		root[i]=1;
		mark[i]=0;
	}
}

void solve(int x)
{
	mark[x]=1;
	if(v[x].size()==0)
	{
		dp[x][1]=w[x];
		dp[x][0]=0;
	}
	int tmp=0,tmp2=0;
	for(int i=0;i<v[x].size();i++)
	{
		if(mark[v[x][i]]==0)
		{
			solve(v[x][i]);
			//上司参加 ，员工不参加 
			tmp+=dp[v[x][i]][0];
			//上司不参加，下属可参加可不参加
			tmp2+=max(dp[v[x][i]][0],dp[v[x][i]][1]);
		}
	}
	dp[x][1]=w[x]+tmp;
	dp[x][0]=tmp2;
}
int main()
{
	int n;
	while(cin>>n&&n!=0)
	{
		init(); 
		for(int i=1;i<=n;i++)
			cin>>w[i];
		for(int i=1;i<n;i++)
		{
			int l,k;
			cin>>l>>k;
			root[l]=0;
			v[k].push_back(l);
		}
		int r=0;
		//得到根节点 
		for(int i=1;i<=n;i++)
			if(root[i]==1)
			{
				r=i;
				break;
			}
		solve(r);
		cout<<max(dp[r][1],dp[r][0])<<endl;
	}
	return 0;
}