54 lines
1.4 KiB
C++
54 lines
1.4 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
const int N = 6010;
|
|||
|
|
|
|||
|
|
struct Edge {
|
|||
|
|
int a, b, c;
|
|||
|
|
const bool operator<(const Edge &t) const {
|
|||
|
|
return c < t.c;
|
|||
|
|
}
|
|||
|
|
} edge[N];
|
|||
|
|
int n;
|
|||
|
|
|
|||
|
|
int cnt[N]; // 配合并查集使用的，记录家族人员数量
|
|||
|
|
int p[N]; // 并查集
|
|||
|
|
int find(int x) {
|
|||
|
|
if (p[x] != x) p[x] = find(p[x]);
|
|||
|
|
return p[x];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
int T;
|
|||
|
|
cin >> T;
|
|||
|
|
while (T--) {
|
|||
|
|
cin >> n;
|
|||
|
|
for (int i = 1; i <= n; i++) p[i] = i, cnt[i] = 1; // 并查集初始化
|
|||
|
|
|
|||
|
|
// 录入n-1条边
|
|||
|
|
int el = n - 1;
|
|||
|
|
for (int i = 0; i < el; i++) {
|
|||
|
|
int a, b, c;
|
|||
|
|
cin >> a >> b >> c;
|
|||
|
|
edge[i] = {a, b, c};
|
|||
|
|
}
|
|||
|
|
// 排序
|
|||
|
|
sort(edge, edge + el);
|
|||
|
|
|
|||
|
|
int res = 0;
|
|||
|
|
for (int i = 0; i < el; i++) {
|
|||
|
|
int a = find(edge[i].a), b = find(edge[i].b), c = edge[i].c;
|
|||
|
|
if (a != b) {
|
|||
|
|
// a集合数量，b集合数量，相乘，但需要减去已经建立的最小生成权这条边
|
|||
|
|
// c是最小的，其它的可以建立最小也得大于c,即c+1
|
|||
|
|
res += (cnt[a] * cnt[b] - 1) * (c + 1);
|
|||
|
|
p[a] = b; // 合并到同一集合
|
|||
|
|
cnt[b] += cnt[a]; // b家族人数增加cnt[a]个,并查集数量合并
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 输出
|
|||
|
|
cout << res << endl;
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|