F - Maximum White Subtree

---

F - Maximum White Subtree

• 看我博客我有没看懂的地方，或者其他疑问，可以加我qq和我交流~我会及时解答
• qq：1244536605（加好友时备注一下 博客 哈）

---

标签

• 换根dp

---

简明题意

• 给定n个节点的树，每个节点的值为0或1.
• 现在需要你对树的每个节点v求出：包含v的联通子图中，节点1的数量减去0的数量，最多是多少。

---

思路

• 假如要求的这个点v是根节点，那么我们可以dp，一个值为1的节点，他的值等于他所有子树中，dp为为正数的和。也就是$$dp[u]=\sum _{v=u的所有儿子且dp[v]为正数}dp[v]+(a[u]==1)$$
• 这里是从下往上的，从叶子节点到根节点，那么可以dfs，或者记忆化搜索。
• 这样下来，我们算出了某个节点的答案，复杂度为O(n)。但一共有n个节点，复杂度就是n方了，无法接受。而这里，我们不需要把每个节点当作根都dp一遍，我们可以换根dp。
• 假设我们先以1号节点为根，dp了一遍。那么dp[i]表示的是节点i的子树中，1的数量和0的数量最大的差值。现在看这个图
  
• 假设现在需要计算包含5号节点的最大差值。那么我们可以把这棵树分成两个部分：
  
• 我们令f(x)表示包含x的最大差值。现在计算f(5)，用x、y标出了这两个部分。我们发现y部分，就等于dp[5]，那么dp[5]>0就累加进f(5)。再看x部分。
• 求x部分，可不可以直接用dp[1]-dp[5]？不可以，因为你不知道y这一部分有没有贡献到f(1)中，即使知道f(5)的正负。比如2和3，它们是1的儿子，那么dp[2]和dp[3]可以确定是否贡献到dp[1]中，也就是只要儿子的dp[]值是正数，那么可以贡献到父节点中。而5和1的关系是孙子。孙子是不一定能贡献到爷爷的。举个例子吧，比如2和5节点之间有100个值为0的，那么y部分就算全部是1也不可能贡献进dp[1].
• 所以我们算x部分的时候，应该从y部分的根节点考虑。也就是从5号节点考虑。我们只能知道5号节点能不能给他的父亲2号节点贡献。考虑到这里，就很容易算了，用f(2)-max(dp[5],0)即可。也就是如果dp[5]>0那么它是贡献进了f(2)的，那么就需要减去。否则没有贡献进去，就不需要减去。
• 所以f的递归式是这样的：
  $$f(u)=dp[u]+max(x,0)$$
  其中
  $$x=f(fa[u])-max(dp[5],0)$$
• 先树形dp，再换根dp
• 这里注意一点，x部分要判断正负以决定是否加入贡献。而x部分，不需要考虑正负，直接贡献。因为，我们求的f(u)一定是包含u的。那么dp[u]就刚好包含u了。

---

注意事项

• 无

---

总结

• 题目询问多个答案，如果每个答案都要以各自的根求出来，那么我们要考虑换根dp

---

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring> 
#include<stack>
#include<map>
#include<queue>
#include<cstdio>	
#include<set>
#include<map>
#include<string>
using namespace std;

const int maxn = 2e5 + 10;

int n, a[maxn], dp[maxn];
vector<int> g[maxn];

int memo[maxn], fa[maxn];
int f(int u, int faa = 0)
{
	if (memo[u] != -2)
		return memo[u];
	if (faa != 0) fa[u] = faa;
	int sum = 0;
	for (auto& v : g[u]) if (v != faa)
		sum += max(f(v, u), 0);
	return memo[u] = sum + (a[u] == 1 ? 1 : -1);
}

int memo2[maxn];
int h(int u)
{
	if (u == 1) return f(1);
	if (memo2[u] != -2)
		return memo2[u];
	int x = h(fa[u]) - max(f(u), 0);
	return memo2[u] = f(u) + max(x, 0);
}

void solve()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
		cin >> a[i], memo[i] = memo2[i] = -2;
	for (int i = 1; i <= n - 1; i++)
	{
		int u, v;
		cin >> u >> v;
		g[u].push_back(v), g[v].push_back(u);
	}

	for (int i = 1; i <= n; i++)
		cout << h(i) << " ";
}
	
int main()
{
	//freopen("Testin.txt", "r", stdin);
	solve();
	return 0;
}