题意
给你一棵树n个节点,每个节点要么是白色(1) 要么是黑色(0) 问每个点输出包含这个点的连通子图的的白色节点的个数黑色节点的个数最大值
解析:
换根dp
第一次dfs:是自底向上随便找一个点作为根 这样就维护了以i为根的子树的最优解
第二次dfs:是自顶向下,因为第一次我们计算出了以i为根的子树最优解,但是少了i上面那一部分的最优解
所以儿子节点j的最优解就是 f[j]=max(0,f[u]-max(0,f[j]))
f[u]-max(0,f[j])是计算i上面那部分的最优解
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10000;
vector<int> G[N];
int f[N];
int a[N];
int n;
void dfs(int u,int fa)
{
for(int i=0;i<G[u].size();i++)
{
int j=G[u][i];
if(j==fa) continue;
dfs(j,u);
f[u]+=max(0,f[j]);
}
}
void dfs2(int u,int fa)
{
for(int i=0;i<G[u].size();i++)
{
int j=G[u][i];
if(j==fa) continue;
f[j]+=max(0,f[u]-max(0,f[j]));
dfs2(j,u);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]==0) f[i]=-1;
else f[i]=1;
}
for(int i=0;i<n-1;i++)
{
int u,v;
scanf("%d %d",&u,&v);
G[u].push_back(v);G[v].push_back(u);
}
dfs(1,-1);
dfs2(1,-1);
for(int i=1;i<=n;i++)
{
cout<<f[i]<<" ";
}
cout<<endl;
}