原题地址
克鲁斯卡尔算法大概分为三部
1、初始化连通分量(并查集完成)
2、判断两个顶点是否在一个连通图中
3、若不在,合并两个图
代码如下:
#include<bits/stdc++.h>
using namespace std;
int f[1000];
int ans;
int t=1;
int find(int x){//并查集模板
if(f[x]==x) return x;
else find(f[x]);
}
struct edge{//边集
int u,v,w;
}p[20000];
bool cmp(edge x,edge y){//重载运算符
return x.w<y.w;
}
int main()
{
int n;
int m=0;
cin>>n;
for(int i=1;i<=n;i++) f[i]=i;//初始化父节点为自己
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
int k=0;
cin>>k;
if(j>i){
m++;
p[m].u=i;p[m].v=j;p[m].w=k;
}
}//存图
sort(p+1,p+m+1,cmp);//按题目要求按边权从小到大排列
for(int i=1;i<=m;i++){
if(find(p[i].u)!=find(p[i].v)){//如果没有同一个祖先节点
ans+=p[i].w;
f[find(p[i].u)]=p[i].v;//合并子树
t++;
if(t==n) break;
}
}
cout<<ans<<endl;
}