CF #627 F. Maximum White Subtree （换根dp）

题目链接

题意：给你一棵树，树上每个节点的值要么为1，要么为0，输出n个数，分别为包含第i 个点在内的联通图中cnt1 - cnt0的最大值是多少？

思路：设dp【i】为包含点 i 再内的cnt1 - cnt0的最大值，那么dp【i】= dp【j】 + a【val】，（j为跟i链接的节点&&dp【j】>0)。这题的关键就是要理解subtree不是子树而是联通图。

#include <bits/stdc++.h>
#define pi pair<int,int>
#define mk make_pair
#define ll long long
#define pb push_back
using namespace std;
const int maxn = 2e5+100,inf = 1e9+10;
vector<int>G[maxn],pre[maxn];
int a[maxn],d[maxn],ans[maxn];
     
void dfs(int u,int fa)
{
    d[u] = a[u];
    for(int i=0;i<G[u].size();i++)
    {
    	int v = G[u][i];
    	if(v == fa)continue;
    	dfs(v,u);
    	if(d[v]>0)d[u] += d[v];
    }
}
     
void dfs1(int u,int fa)
{
    d[u] = a[u];
    pre[u].pb(d[u]);
    for(int i=0;i<G[u].size();i++)
    {
    	int v = G[u][i];
    	if(d[v]>0)d[u] += d[v];
    	pre[u].pb(d[u]);
    }
   	ans[u] = d[u];
    int x = 0;
    for(int i=G[u].size()-1;~i;i--)
    {
    	int v = G[u][i];
    	d[u] = pre[u][i] + x;
    	if(d[v] > 0)x += d[v];
    	if(v!=fa)dfs1(v,u);
    }
}
     
int main()
{
   	int n;
   	scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
    	scanf("%d",a+i);
    	if(a[i] == 0)a[i] = -1;
    }
    for(int i=1;i<n;i++)
    {
    	int u,v;
    	scanf("%d%d",&u,&v);
    	G[u].pb(v);
    	G[v].pb(u);
    } 
    dfs(1,0);
    dfs1(1,0);
    for(int i=1;i<=n;i++)cout<<ans[i]<<' ';
    return 0;
}