【Week13作业 D】TT的苹果树【树型dp】

题意：

已知树上共有 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 结束。
每组数据输出一个整数，代表所选苹果快乐值总和的最大值。

---

思路：

存储树可以使用节点-孩子的结构,int w为节点,vector child为孩子。
令f[i][j]是以i为根的子树的最大快乐值，j=0为不选择i节点，j=1为选择i节点。则f[i][0]=孩子节点∑max(f[x][0],f[x][1])，f[i][1]=孩子节点∑f[x][0]+tree[i].w。
根据输入找出树根，然后从树根开始递归求解，答案就是max(f[root][0],f[root][1])。

---

总结：

一道树型dp题目，很好的体现了树型dp的递归特点。

---

代码：

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

const int MAXN=6005;
//树
struct point
{
	int w;
	vector<int> child;
}tree[MAXN]; 
bool root[MAXN];
//dp
int f[MAXN][2];
void dp(int i)
{
	f[i][0]=0,f[i][1]=tree[i].w;
	for(int j=0;j<tree[i].child.size();j++)
	{
		int x=tree[i].child[j];
		dp(x);
		f[i][0]=f[i][0]+max(f[x][1],f[x][0]);
		f[i][1]=f[i][1]+f[x][0];
	}
}
int ans;
int main()
{
	int n;
	while(cin>>n)
	{
		if(n==0)
		{
			int n1;
			cin>>n1;
			break;
		}
		for(int i=1;i<=n;i++)
			cin>>tree[i].w;
		for(int i=1;i<=n;i++)
			root[i]=1,tree[i].child.clear();
		for(int i=1;i<=n-1;i++)
		{
			int l,k;
			cin>>l>>k;
			tree[k].child.push_back(l);
			root[l]=0;
		}
		int rt;	//树根 
		for(int i=1;i<=n;i++)
			if(root[i])
				rt=i;
		dp(rt);
		ans=max(f[rt][0],f[rt][1]);
		cout<<ans<<endl;
	}
}