E - Minimize Sum of Distances（树上问题）

E - Minimize Sum of Distances（树上问题）

题目链接

思路

这题也是典中典，我们可以将$C_i$看做点的权重，开一个数组$siz[i]$代表以点$i$为根的子树权重，首先我们可以跑一遍dfs预处理出$siz$数组，同时计算一下$i$为$1$时$f(1)$的大小，然后怎么计算其他$f(i)(i\ne 1)$，假设1的子节点为$a$，那么当1转移至$a$时，以$a$为根的子树的节点与$a$之间的边都少了一条(相对于节点1)，那么$f$也就减小了$siz[a]$，以$a$为根节点的子树的节点以外的节点，与$a$之间的边都多了一条，$f$也就增加了$siz[1]-siz[a](其他节点的总权重)$，所以$f(a)=f(1)+siz[1]-2\times 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;
}