【BZOJ3302】[Shoi2005]树的双中心【DFS】【TreeDP】

【题目链接】

考虑暴力做法，我们可以枚举删掉某条边，然后在两个子树里找重心，统计答案即可，O(n^2)的。

发现树高最多100，并且发现每次转移只可能向着权值和更大的地方移动，那么我们可以记录出每个节点权值和最大的儿子。但是如果删掉一条边的时候，把这个儿子给归到另一个子树里了，所以我们还得记录次大权值和的儿子。

复杂度O(nh)。

设sum[x]表示x的子树中的权值和。

设res[x]表示在x的子树中，将x作为中心的答案。res[x]是可以递推转移的，res[x] = ∑(res[v] + sum[v])。

设son[x]和bro[x]分别表示x的权值和最大和次大的儿子。这个很好维护。

我们从根节点root递归，假设到了点x，枚举到儿子v。

我们将(u, v)这条边切掉，那么出现两棵树，一个包含root，一个包含v。

接下来在两棵树里分别移动点，找中心。

对于包含root的树，初始值root是中心，那么答案为res[root] - res[v] - depth[v] * sum[v]（仔细推推就是这样）。

对于包含v的树，初始值v是中心，那么答案为res[v] 。

移动的时候，也是递推计算答案。

假设当前树的根节点为rt，点u的答案为k，现在要转移到u的儿子v，那么答案可以O(1)计算，为k + sum[rt] - 2 * sum[v]（仔细推推...）。

最后用两棵树里算出的值加起来更新答案即可。

此题比较好，因为是三倍经验题。

/* Pigonometry */
#include <cstdio>
#include <algorithm>

using namespace std;

typedef long long LL;

const int maxn = 50005, inf = 0x3f3f3f3f;

int n, head[maxn], cnt, pre[maxn], depth[maxn], son[maxn], bro[maxn];
LL sum[maxn], res[maxn], ans;

struct _edge {
	int v, next;
} g[maxn << 1];

inline int iread() {
	int f = 1, x = 0; char ch = getchar();
	for(; ch < '0' || ch > '9'; ch = getchar()) f = ch == '-' ? -1 : 1;
	for(; ch >= '0' && ch <= '9'; ch = getchar()) x = x * 10 + ch - '0';
	return f * x;
}

inline void add(int u, int v) {
	g[cnt] = (_edge){v, head[u]};
	head[u] = cnt++;
}

inline void dfs(int x) {
	for(int i = head[x]; ~i; i = g[i].next) if(g[i].v ^ pre[x]) {
		depth[g[i].v] = depth[x] + 1;
		pre[g[i].v] = x;
		dfs(g[i].v);
		sum[x] += sum[g[i].v];
		res[x] += res[g[i].v] + sum[g[i].v];
		if(!son[x] || sum[g[i].v] > sum[son[x]]) bro[x] = son[x], son[x] = g[i].v;
		else if(!bro[x] || sum[g[i].v] > sum[bro[x]]) bro[x] = g[i].v;
	}
}

int cut;

inline void center(LL &ret, int root, int x, LL k) {
	ret = min(ret, k);
	int v = son[x];
	if(v == cut || sum[bro[x]] > sum[son[x]]) v = bro[x];
	if(!v) return;
	center(ret, root, v, k + sum[root] - 2 * sum[v]);
}

inline void solve(int x) {
	for(int i = head[x]; ~i; i = g[i].next) if(g[i].v ^ pre[x]) {
		cut = g[i].v;
		LL gx = inf, gy = inf;
		for(int j = x; j; j = pre[j]) sum[j] -= sum[cut];
		center(gx, 1, 1, res[1] - res[cut] - depth[cut] * sum[cut]);
		center(gy, cut, cut, res[cut]);
		ans = min(ans, gx + gy);
		for(int j = x; j; j = pre[j]) sum[j] += sum[cut];
		solve(g[i].v);
	}
}

int main() {
	n = iread();
	for(int i = 1; i <= n; i++) head[i] = -1; cnt = 0;

	for(int i = 1; i < n; i++) {
		int u = iread(), v = iread();
		add(u, v); add(v, u);
	}
	for(int i = 1; i <= n; i++) sum[i] = iread();

	ans = inf;
	dfs(1);
	solve(1);
	printf("%lld\n", ans);
	return 0;
}