D - TT 的苹果树

题目

样例输入

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

样例输出

5

 

思路

就是那个“没有上司的舞会”的问题。

首先通过 is_son 布尔类型的数组选择出来这棵树的根，申请 vector<int> 类型的数组，存储每个节点的孩子。

从根开始进行递归，状态转移：如果孩子的状态转移没有进行完，就先对孩子进行状态转移。

 f [ i ][ 0 / 1 ] 表示以 i 为节点的子树所能得到的最大的欢乐值， 

f [ i ][ 0 ] 表示以 i 为子树且未选择 i 号节点； f [ i ][ 1 ] 表示以 i 为子树且选择了 i 号节点。

状态转移过程

①不摘 i 节点的果子，其孩子节点的果子可摘可不摘；②摘 i 节点的果子，其孩子节点的果子必然不能摘。

 

代码

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

vector<int> son[6010];
int w[6010], f[6010][2];
bool is_son[6010], vis[6010];
int n, a, b, father;

void dfs(int x)
{
	if (vis[x]) return;

	f[x][0] = 0;
	f[x][1] = w[x];
	for (vector<int>::iterator it = son[x].begin(); it != son[x].end(); it++)
	{
		dfs(*it);
		f[x][0] += max(f[*it][0], f[*it][1]);
		f[x][1] += f[*it][0];
	}

	vis[x] = 1;
}

int main()
{
	while (cin >> n)
	{
		if (n == 0) break;
		memset(is_son, 0, sizeof is_son);
		memset(vis, 0, sizeof vis);
		memset(f, 0, sizeof f);
		for (int i = 1; i <= n; i++)
			cin >> w[i];

		for (int i = 1; i < n; i++)
		{
			cin >> a >> b;
			son[b].push_back(a);
			is_son[a] = 1;
		}


		for(int i=1;i<=n;i++)
			if (!is_son[i])
			{
				father = i;
				break;
			}
		dfs(father);
		cout << max(f[father][0], f[father][1]) << endl;
	}

}