parent
319de2c540
commit
aaec68c0f4
@ -0,0 +1,54 @@
|
||||
#include <bits/stdc++.h>
|
||||
using namespace std;
|
||||
const int N = 200010, M = N << 1;
|
||||
|
||||
int n, a[N], f1[N], f2[N];
|
||||
int head[N << 1], nxt[N << 1], to[N << 1], tot = 0;
|
||||
void add(int u, int v) {
|
||||
to[++tot] = v;
|
||||
nxt[tot] = head[u];
|
||||
head[u] = tot;
|
||||
}
|
||||
|
||||
void dfs1(int u, int fa) {
|
||||
if (a[u])
|
||||
f1[u] = 1;
|
||||
else
|
||||
f1[u] = -1;
|
||||
|
||||
for (int i = head[u]; i; i = nxt[i]) {
|
||||
int v = to[i];
|
||||
if (v == fa)
|
||||
continue;
|
||||
dfs1(v, u);
|
||||
f1[u] += max(f1[v], 0);
|
||||
}
|
||||
}
|
||||
|
||||
void dfs2(int u, int fa) {
|
||||
for (int i = head[u]; i; i = nxt[i]) {
|
||||
int v = to[i];
|
||||
if (v == fa)
|
||||
continue;
|
||||
f2[v] = max(f2[u] + f1[u] - max(f1[v], 0), 0);
|
||||
dfs2(v, u);
|
||||
}
|
||||
}
|
||||
int main() {
|
||||
cin >> n;
|
||||
|
||||
for (int i = 1; i <= n; i++) cin >> a[i];
|
||||
|
||||
for (int i = 1; i <= n - 1; i++) {
|
||||
int a, b;
|
||||
cin >> a >> b;
|
||||
add(a, b), add(b, a);
|
||||
}
|
||||
|
||||
dfs1(1, 1);
|
||||
|
||||
dfs2(1, 1);
|
||||
for (int i = 1; i <= n; i++) printf("%d ", f1[i] + f2[i]);
|
||||
|
||||
return 0;
|
||||
}
|
Loading…
Reference in new issue