最小生成树的两种算法总结（克鲁斯塔尔，prim）

克鲁斯塔尔：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
struct node
{
int from,to,va; //记录起止点和边的权值
}a[100010];
int f[100010];
bool cmp(node x,node y)
{
return x.va<y.va; //从小到大排序
}
int find(int x)
{
if(x!=f[x])
f[x]=find(f[x]);
return f[x];
}
bool join(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
f[fx]=fy;
return true;
}
return false;
}
int main()
{
int n,m;
while(~scanf("%d",&m)&&m)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
f[i]=i; //初始化
for(int i=1;i<=m;i++)
scanf("%d %d %d",&a[i].from,&a[i].to,&a[i].va);
sort(a+1,a+1+m,cmp);
int ant=0;
int num=0;
for(int i=1;i<=m;i++)
{
if(join(a[i].from,a[i].to)) //之前没有可以使他们相通的就加上边的权值
{
ant+=a[i].va;
num++; //多一条边 num++
}
if(num==n-1) // n个点n-1条边，满足退出
break;
}
num=0;
for(int i=1;i<=n;i++)
{
if(f[i]==i)
num++;
if(num>1) //判断是否只有一个根节点
break;
}
if(num>1) //统计数据不足以保证畅通,根节点不唯一
printf("?\n");
else
printf("%d\n",ant);
}
return 0;
}

prim：
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
int map[105][105];
int prim(int n)
{
int vis[105]={0},dis[105],ans=0;
memset(dis,inf,sizeof(dis));//初始化为无穷大
dis[1]=0;//随意选定一个点开始
for(int i=1;i<=n;++i)//循环加入n个顶点
{
int k=-1,tp=inf;
for(int j=1;j<=n;++j)
//查找距离已连接的部分最近的顶点
{
if(!vis[j]&&dis[j]<tp)
{
k=j;tp=dis[j];
}
}
vis[k]=1;ans+=tp;//统计和标记
for(int j=1;j<=n;++j)
//更新未选中的点和已经连接的部分之间的最短距离
{
if(!vis[j]&&map[k][j]<dis[j])
{
dis[j]=map[k][j];
}
}
}
return ans;
}
int main()
{
int n,m,a,b,c;
while(scanf("%d",&n)&&n)
{
memset(map,inf,sizeof(map));
m=n*(n-1)/2;//边的数量
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
map[a][b]=map[b][a]=c;//无向边
}
printf("%d\n",prim(n));
}
return 0;
}