/*Kruskal*/
#include <iostream>
#include <algorithm>
#include <fstream>
using namespace std;
typedef struct _edge
{
int u;
int v;
int weight;
}edge;
//var
fstream fin;
//function
void Kruskal(edge *e,int m,int n);
int main()
{
fin.open("1010.txt",ios::in);
int n,m;
fin>>n>>m;
edge *e=new edge[m];
for(int i=0;i<m;i++)
{
fin>>e[i].u>>e[i].v>>e[i].weight;
e[i].u--;
e[i].v--;
}
Kruskal(e,m,n);
system("pause");
return 0;
}
int cmp(const void *a,const void *b)
{
return (*(edge *)a).weight-(*(edge *)b).weight;
}
//kruskal算法
void Kruskal(edge *e,int m,int n)
{
qsort(e,sizeof(e),sizeof(e[0]),cmp);
int *vertices=new int[n];//计算每个顶点的的分支
memset(vertices,0,n*sizeof(int));
int parts=0;
int u,v;
for(int i=0;i<m;i++)
{
u=e[i].u;
v=e[i].v;
if(!vertices[u]&&!vertices[v])
{
parts++;
vertices[u]=parts;
vertices[v]=parts;
cout<<u<<" "<<v<<endl;
}
else
{
if(vertices[u]&&!vertices[v])
{
vertices[v]=vertices[u];
cout<<u<<" "<<v<<endl;
}
else
{
if(vertices[v]&&!vertices[u])
{
vertices[u]=vertices[v];
cout<<u<<" "<<v<<endl;
}
else if(vertices[u]!=vertices[v])
{
//找到最小的分支
cout<<u<<" "<<v<<endl;
int temp1=vertices[u];//小的parts
int temp2=vertices[v];
if(vertices[u]>vertices[v])
{
temp1=vertices[v];
temp2=vertices[u];
}
//更新分支
for(int j=0;j<n;j++)
{
if(vertices[j]==temp2)
vertices[j]=temp1;
else if(vertices[j]>temp2)
vertices[j]--;
}
parts--;
}
}
}
}
}
Kruskal算法求最小生成树
最新推荐文章于 2022-12-17 19:56:50 发布