这题和前一题都属于并查集,但是又有点不同,是给了你所有的边,然后求连通图的最小生成树,连通图中不能有环。
这题的方法是kruskal算法+并查集,按照边的权值排序,然后从小到大遍历,如果两点之间已经联通,则不取,如果不联通,则取(详见kruskal算法求SMT),贴代码。
#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int pre[105];
struct edge
{
int a,b,e;
}p[5000];
bool cmp(struct edge a,struct edge b)
{
return a.e<b.e;
}
int Find(int x)
{
return pre[x]==x?x:pre[x]=Find(pre[x]);
}
int main()
{
int n;
while(cin>>n&&n)
{
for(int i=0;i<n*(n-1)/2;i++)
{
scanf("%d%d%d",&p[i].a,&p[i].b,&p[i].e);
}
for(int i=1;i<=n;i++) pre[i]=i;
sort(p,p+n*(n-1)/2,cmp);
int sum=0;
for(int i=0;i<n*(n-1)/2;i++)
{
int fa=Find(p[i].a);
int fb=Find(p[i].b);
if(fa!=fb)
{
sum+=p[i].e;
pre[fa]=fb;
}
}
printf("%d\n",sum);
}
return 0;
}