https://www.acwing.com/problem/content/description/348/
题意:给定一棵N个节点的树,要求增加若干条边,
把这棵树扩充为完全图,并满足图的唯一最小生成树仍然是这棵树。
求增加的边的权值总和最小是多少。
思路 克鲁斯卡尔变种
使用并查集
把边排序,如果俩端不在同一集合,连边,权重为w
要成为完全图,其他边权重为w+1,边数为:俩个集合点数乘积
另外开s数组记录
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=6666;
struct node{
int a,b,c;
bool operator<(const node& t){
return c<t.c;
}
};
node edge[6666];
int f[N],s[N];
int get(int x){
return x==f[x]?x:f[x]=get(f[x]);
}
int main(){
int t;
cin>>t;
while(t--){
int n;
cin>>n;
for(int i=1;i<=n;i++){
f[i]=i,s[i]=1;
}
for(int i=0;i<n-1;i++){
cin>>edge[i].a>>edge[i].b>>edge[i].c;
}
long long int res=0;
sort(edge,edge+n-1);
for(int i=0;i<n-1;i++){
int x=edge[i].a,y=edge[i].b,w=edge[i].c;
int fx=get(x);
int fy=get(y);
if(fx==fy) continue;
res+=1ll*(s[fy]*s[fx]-1)*(w+1);
f[fx]=fy;
s[fy]+=s[fx];
}
cout<<res<<endl;
}
return 0;
}