Maximum White Subtree (树形dp)

Maximum White Subtree

题目链接

大致题意:

给定一棵n个节点无根树，每个节点u有一个颜色a[u],若a[u]为0则u是黑点,若a[u]为1则u是白点
对于每个节点u，选出一个包含u的连通子图，设子图中白点个数为cnt1，黑点个数为cnt2，请最大化cnt1−cnt2.并输出这个值.

---

解题思路:

树形dp

对于一个节点来说,最大值的出现可能是它的其中一颗子树的最大值,或者是多颗子树最大值相加的结果.为了方便,我们定义以1为根,定义a[i]表示以i为根的子树的最大值,这里我们用一个dfs就可以求出来,下面就是考虑情况了.

既然是一棵树,并且我们已经求出以每个点为根的子树的最大值了,剩下我们就要判断当前点u需不需要与父节点做连接,也就是我们需要知道u点的父节点之上的子树的最大值

举例来说,当u是6时,我们需要加上蓝色圈起来的以2为根的最大值,这个最大值怎么求让人很麻爪

我先说做法:对于点u,v时u的子节点

如果a[v]>=0 ,以v为根节点的子树的最大值时包含在以u为根节点的子树的最大值里的,这是我们前面dfs的结果,所以,对于子节点v,我们要选择max(以v为根节点的子树的最大值,以u为根节点的子树的最大值),即a[v]=max(a[v],a[u]) .

解释:我们从上往下进行递归,上面的操作,已经帮我们实现了求蓝色圈起来的以2为根的最大值.

如果a[v]<0 ,此时以v为根节点的子树的最大值时不包含在以u为根节点的子树的最大值里的,因为前面当a[v]>0时,我们才累加到a[u]上,所以,对于子节点v,我们要选择max(以v为根节点的子树的最大值,以v为根节点的子树的最大值+以u为根节点的子树的最大值),即a[v]=max(a[v],a[v]+a[u])

所以,最后a[i]的值就表示包含点i的联通块,白点个数-黑点个数的最大值

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
int n, m;
int a[N];
vector<int>e[N];
void dfs1(int u, int fa) {   //求以u为根的最大值
	for (auto& v : e[u]) {
		if (v == fa)continue;
		dfs1(v, u);
		if (a[v] > 0)a[u] += a[v];  //这里打个标记
	}
}
void dfs2(int u, int fa) {
	for (auto& v : e[u]) {
		if (v == fa)continue;
		if (a[v] >= 0)a[v] = max(a[v], a[u]);
		else a[v] = max(a[v], a[v] + a[u]);
		dfs2(v, u);
	}
}
int main(void)
{
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) {
		scanf("%d", &a[i]);
		if (!a[i])a[i] = -1;
	}
	for (int i = 1; i < n; ++i) {
		int u, v; scanf("%d%d", &u, &v);
		e[u].push_back(v); e[v].push_back(u);
	}
	dfs1(1, 0);
	dfs2(1, 0);
	for (int i = 1; i <= n; ++i)
		printf("%d%c", a[i], i == n ? '\n' : ' ');
	return 0;
}

---