Codeforces Round #627 (Div. 3) F - Maximum White Subtree(深度优先搜索)

Posted kanoon

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Codeforces Round #627 (Div. 3) F - Maximum White Subtree(深度优先搜索)相关的知识,希望对你有一定的参考价值。

题意:

有n个点n-1条边的数,问每个点所在子树中白黑点数目最大差。

思路:

白点先由下至上汇集,后由上至下分并。

#include <bits/stdc++.h>
using namespace std;

const int M=220000;

vector<vector<int>> e(M);

int n,a[M];

void dfs1(int u,int pre){
    for(int v:e[u]){
        if(v!=pre){
            dfs1(v,u);
            a[u]+=max(0,a[v]);
        }
    }
}

void dfs2(int u,int pre){
    for(int v:e[u]){
        if(v!=pre){
            a[v]+=max(0,a[u]-max(0,a[v]));
            dfs2(v,u);
        }
    }
}

int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
        if(a[i]==0) a[i]=-1;
    }
    for(int i=0;i<n-1;i++){
        int u,v;cin>>u>>v;
        --u,--v;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs1(0,-1);
    dfs2(0,-1);
    for(int i=0;i<n;i++) cout<<a[i]<< ;
    return 0;
}

 

以上是关于Codeforces Round #627 (Div. 3) F - Maximum White Subtree(深度优先搜索)的主要内容,如果未能解决你的问题,请参考以下文章

Codeforces Round #627 (Div. 3) 补题

Codeforces Round #627 (Div. 3)

Codeforces Round #627 (Div. 3)(A--D)

Codeforces Round #627 (Div. 3) D. Pair of Topics(二分/直接遍历)

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

Codeforces Round #627 (Div. 3) F - Maximum White Subtree(深度优先搜索)