这道题感觉思路非常巧妙, 我是看了别人的博客才想明白的。
这里用到了并查集, 以根节点为中心城市, 然后把边从大到小排序, 每次的当前的边即为容量,
因为是目前的最小值, 然后去算总的容量, 每次选容量大的点作为新的根节点, 也就是新的
中心城市。细节见代码。
#include<cstdio>
#include<algorithm>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;typedef long long ll;
const int MAXN = 212345;
struct node
{ll u, v, w;bool operator < (const node& rhs) const{return w > rhs.w;}
}Edge[MAXN];
ll sum[MAXN], cnt[MAXN];
int f[MAXN], n;int find(int x)
{if(f[x] != x)f[x] = find(f[x]);return f[x];
}int main()
{while(~scanf("%d", &n)){REP(i, 1, n + 1) sum[i] = 0, cnt[i] = 1, f[i] = i;REP(i, 0, n - 1) scanf("%lld%lld%lld", &Edge[i].u, &Edge[i].v, &Edge[i].w);sort(Edge, Edge + n - 1);ll ans = 0;REP(i, 0, n - 1){int u = find(Edge[i].u), v = find(Edge[i].v), w = Edge[i].w;ll sumu = sum[u] + cnt[v] * w, sumv = sum[v] + cnt[u] * w;if(sumu < sumv) f[u] = v, ans = sumv, cnt[v] += cnt[u], sum[v] = sumv;else f[v] = u, ans = sumu, cnt[u] += cnt[v], sum[u] = sumu;} printf("%lld\n", ans);}return 0;
}