Codeforces Round #627 (Div. 3) F. Maximum White Subtree ###K ###K ###K ###K//K

题目链接：https://codeforces.ml/contest/1324/problem/F

题意：给定一棵树，每个点  为1的时白色点，为0的是黑色点，问每一个点  包含该点在内的子树的最大的  cnt白-cnt黑为多少

思路：树形dp 用上换根法， 首先以1为根跑一遍dfs 求出 所有的点的子树的最大值的， 此时只有1的值是确定的最大的，其他的点都是只有子树是最大的，上面的部分还没有计算

然后再跑一遍dfs 开始换根     每一个点，要么以子树的为最大的，要么是以除子树之外的上部为最大的，要么是子树加上上部为最大的

每次跑的时候，上面的根已经处理好了，所以跑一遍就可以得到所有的点的答案   只需讨论一下 根的dp[u] 内有无包含cnt[v]就可以了  

cnt[v]>0 肯定对根由贡献，小于0肯定之前没被加入到根的贡献中  易错点在于只有ans[fa]>cnt[u]的才会去考虑上子树的贡献

 

当要求的是时树上所有结点的问题 并且 子节点可以由父节点的关系推来时就采用换根dp

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
const int maxn =2e5+10;
const int mod=1e9+7;
int a[maxn];
int cnt[maxn];
int dp[maxn];
vector<int>E[maxn];



void dfs(int u,int fa)
{
    cnt[u]=a[u];
    for(auto &v:E[u])
    {
        if(v==fa)
            continue;
        dfs(v,u);
        cnt[u]+=max(0,cnt[v]);

    }
}

void dfs2(int u,int fa)
{
    for(auto &v:E[u])
    {
        if(v==fa)
            continue;
        if(cnt[v]>=0)
        {
            dp[v]=cnt[v]+max(0,dp[u]-cnt[v]);
        }
        else
        {
            dp[v]=cnt[v]+max(0,dp[u]);
        }
        dfs2(v,u);
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        if(!a[i])
            a[i]=-1;
    }
    for(int i=1;i<n;i++)
    {
        int u,v;
        cin>>u>>v;
        E[u].pb(v);
        E[v].pb(u);
    }
    dfs(1,0);
    dp[1]=cnt[1];
    dfs2(1,0);
    for(int i=1;i<=n;i++)
    {
        cout<<dp[i]<<" ";
    }






}