题意:
给一棵树的图,每个节点为
或
分别代表白和黑,对每个点输出包含这个点的连通子图的的白色节点的个数黑色节点的个数最大值。
很裸的换根
,第一遍
随便找一个点作为树根,第二次
开始转移。
AC代码:
const int N = 2e5 + 5;
vector<int> edge[N];
int a[N], ans[N];
int n;
void dfs(int root, int par)
{
for (auto i : edge[root])
{
if (i == par)
continue;
dfs(i, root);
if (a[i] > 0)
a[root] += a[i];
}
}
void dfs2(int root, int par, int pv)
{
ans[root] = a[root] + pv;
for (auto i : edge[root])
{
if (i == par)
continue;
dfs2(i, root, max(0, ans[root] - max(0, a[i])));
}
}
int main()
{
sd(n);
rep(i, 1, n)
{
sd(a[i]);
a[i] *= 2;
a[i] -= 1;
}
rep(i, 1, n - 1)
{
int x, y;
sdd(x, y);
edge[x].pb(y);
edge[y].pb(x);
}
dfs(1, -1);
dfs2(1, -1, 0);
rep(i, 1, n)
printf("%d%c", ans[i], i == n ? '\n' : ' ');
return 0;
}