标签:names 原来 printf kruskal 输入 algo using i++ round
题目链接:
【简化版题意】给定一棵N个节点的树,要求增加若干条边,把这棵树扩充为完全图,并满足图的唯一最小生成树仍然是这棵树。求增加的边的权值总和最小是多少。
输入格式
本题为多组数据~
第一行t,表示有t组测试数据
对于每组数据
第一行N,表示水龙头的个数(当然也是OIER的个数);
2到N行,每行三个整数X,Y,Z;表示水龙头X和水龙头Y有一条长度为Z的小道输出格式
对于每组数据,输出一个整数,表示修建的所有道路总长度的最短值。
样例输入
2 3 1 2 2 1 3 3 4 1 2 3 2 3 4 3 4 5样例输出
4 17数据范围与约定
- 每个测试点最多10组测试数据
50% n<=1500;
100% n<=6000
100% z<=100样例解释
第一组数据,在2和3之间修建一条长度为4的道路,是这棵树变成一个完全图,且原来的树依然是这个图的唯一最小生成树.
代码:
1 #include <cstdio> 2 #include <algorithm> 3 #include <cctype> 4 #include <cstring> 5 #include <iostream> 6 #include <queue> 7 using namespace std; 8 9 const int N=6010; 10 long long ans=0; 11 int fa[N],size[N],T,n; 12 struct E{int x,y,z;}e[N]; 13 bool operator<(const E &n1,const E &n2){return n1.z<n2.z;} 14 15 inline int get(int x){ 16 if(x==fa[x]) return x; return fa[x]=get(fa[x]); 17 } 18 int main() 19 { 20 scanf("%d",&T); 21 while(T--) 22 { 23 scanf("%d",&n); 24 for(int i=1 ; i<n ; i++) scanf("%d %d %d",&e[i].x,&e[i].y,&e[i].z); 25 for(int i=1 ; i<=n ; i++) fa[i]=i,size[i]=1; 26 sort(e+1,e+n); ans=0; 27 for(int i=1 ; i<n ; i++) 28 { 29 int x=get(e[i].x),y=get(e[i].y); 30 if(x==y) continue; 31 fa[x]=y; ans+=(long long)(size[x]*size[y]-1)*(e[i].z+1); 32 size[y]+=size[x]; 33 } 34 printf("%lld\n",ans); 35 } 36 37 return 0; 38 }
标签:names 原来 printf kruskal 输入 algo using i++ round
原文地址:https://www.cnblogs.com/wmq12138/p/10340259.html