E - Minimize Sum of Distances(树上问题)
思路
这题也是典中典,我们可以将 C i C_i Ci看做点的权重,开一个数组 s i z [ i ] siz[i] siz[i]代表以点 i i i为根的子树权重,首先我们可以跑一遍dfs预处理出 s i z siz siz数组,同时计算一下 i i i为 1 1 1时 f ( 1 ) f(1) f(1)的大小,然后怎么计算其他 f ( i ) ( i ≠ 1 ) f(i)(i \neq1) f(i)(i=1),假设1的子节点为 a a a,那么当1转移至 a a a时,以 a a a为根的子树的节点与 a a a之间的边都少了一条(相对于节点1),那么 f f f也就减小了 s i z [ a ] siz[a] siz[a],以 a a a为根节点的子树的节点以外的节点,与 a a a之间的边都多了一条, f f f也就增加了 s i z [ 1 ] − s i z [ a ] ( 其他节点的总权重 ) siz[1]-siz[a](其他节点的总权重) siz[1]−siz[a](其他节点的总权重),所以 f ( a ) = f ( 1 ) + s i z [ 1 ] − 2 × s i z [ a ] f(a)=f(1)+siz[1]-2\times siz[a] f(a)=f(1)+siz[1]−2×siz[a] ,其他点也这样转移即可
代码
开始因为ans开小了,调了巨长时间,脑*了
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
using ll = long long;
int n;
int a[N];
int h[N], e[N * 2], ne[N * 2], idx = 1;
ll siz[N];
ll res = 0;
ll ans = 1ll << 62;
inline void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u, int fa) {
siz[u] = a[u];
for (int i = h[u]; i; i = ne[i]) {
int j = e[i];
if (j == fa) continue;
dfs(j, u);
siz[u] += siz[j];
res += siz[j];
}
}
void dfs1(int u, int fa, ll sum) {
ans = min(sum, ans);
for (int i = h[u]; i; i = ne[i]) {
int j = e[i];
if (j == fa) continue;
dfs1(j, u, sum + siz[1] - 2 * siz[j]);
}
}
int main() {
cin >> n;
for (int i = 1; i < n; i++) {
int l, r;
cin >> l >> r;
add(l, r), add(r, l);
}
for (int i = 1; i <= n; i++) cin >> a[i];
dfs(1, 0);
dfs1(1, 0, res);
cout << ans << endl;
return 0;
}