P2986 [USACO10MAR]Great Cow Gathering G(换根dp)

这题和P3478 [POI2008]STA-Station本质上是一样的,都是用换根dp进行解决
贴一个换根dp分析

之后我们就可以通过一次自上而下和自下而上的换根dp,进行求解
Size表示子树大小
dp[i]表示当前i到子树整个连通块的所有节点的长度
d[i]表示当前以i为根节点到其它节点的总长度
a[i]表示在i节点有多少头牛

#include<bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
using namespace std;
const int maxn = 1e5 + 5;
vector< pii >g[maxn];
ll a[maxn],Size[maxn],dp[maxn],d[maxn];
ll sum,n,ans = 1e18;
//Size表示子树大小
//dp[i]表示当前i到子树整个连通块的所有节点的长度
//d[i]表示当前以i为根节点到其它节点的总长度
//a[i]表示在i节点有多少头牛 
void dfs1(int u, int fa){
	Size[u] = a[u];
	for(int i = 0; i < g[u].size(); i++){
		int j = g[u][i].first;
		int w = g[u][i].second;
		if(j == fa)continue;
		dfs1(j, u);
		Size[u] += Size[j];
		dp[u] += dp[j] + Size[j]*w;
	}
}
void dfs2(int u, int fa){
	for(int i = 0; i < g[u].size(); i++){
		int j = g[u][i].first;
		int w = g[u][i].second;
		if(j == fa)continue;
		d[j] = d[u] + sum*w - 2*Size[j]*w;
		ans = min(ans, d[j]);
		dfs2(j, u);
	}
}
int main(){
	cin >> n;
	for(int i = 1; i <= n; i++)cin >> a[i],sum += a[i];
	for(int i = 1; i < n; i++){
		int u,v,w;
		cin >> u >> v >> w;
		g[u].push_back(make_pair(v,w));
		g[v].push_back(make_pair(u,w));
	}
	dfs1(1, 0);
	d[1] = dp[1];
	ans = min(ans, d[1]);
	dfs2(1, 0);
	cout << ans << '\n';
}