该算法基本思想:首先按照边的权值大小进行从下到大的排序,每次从剩下的边中选择权值较小且边的两个顶点不在同一个集合内的边(就是不会产生回路的边),加入到生成树中,直到加入了n-1条边为止。
代码模板:
#include <stdio.h>
#include <string.h>
struct edge
{
int u;
int v;
int w;
};
struct edge e[1000010];
int n,m,sum,count;
int f[100010];
void quicksort(int left,int right)
{
int i,j;
struct edge t;
if(left>right)
return ;
i=left;
j=right;
while(i!=j)
{
while(e[j].w>=e[left].w&&i<j)//先从右边找个比基准数小的才停止
j--;
while(e[left].w>=e[i].w&&i<j)//从左边找个比基准数大的才停止
i++;
if(i<j)//交换
{
t=e[i];
e[i]=e[j];
e[j]=t;
}
}
//最终将基准数归位,将left与i互换
t=e[left];
e[left]=e[i];
e[i]=t;
quicksort(left,i-1);
quicksort(i+1,right);
return ;
}
int getf(int v)//并查集寻找祖先的函数
{
if(f[v]==v)
return v;
else
{
f[v]=getf(f[v]);
return f[v];
}
}
int merge(int v,int u)//并查集合并两子集合的函数
{
int t1,t2;
t1=getf(v);
t2=getf(u);
if(t1!=t2)
{
f[t2]=t1;
return 1;
}
return 0;
}
int main()
{
int i,j;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(i=1;i<=m;i++)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
memset(f,0,sizeof(f));
quicksort(1,m);//按照权值从小到大对边进行快速排序
for(i=1;i<=n;i++)//并查集初始化
f[i]=i;
for(i=1;i<=m;i++)
{
if(merge(e[i].u,e[i].v))//如果目前没有连通,则选用这条边
{
count++;
sum+=e[i].w;
}
if(count==n-1)
break;
}
printf("%d\n",sum);
}
return 0;
}