poj 1251(Kruskal)

最新推荐文章于 2019-05-09 11:06:46 发布

weiwei2012start

最新推荐文章于 2019-05-09 11:06:46 发布

阅读量639

点赞数

CC 4.0 BY-SA版权

文章标签： struct

本文链接：https://blog.youkuaiyun.com/sxwup/article/details/7871730

本文介绍了一种使用C++实现的Kruskal算法模板，该算法用于解决最小生成树问题。通过边的权重进行排序，并利用并查集来避免形成环路，最终找到权重最小的生成树。

//Kruskal 直接套的模板

#include<stdio.h>
#include<algorithm>
using namespace std;
typedef struct
{
int x,y,w;
}Edge;

Edge edge[30*30];
int edgenum,minCost,n;
int father[30];
void makeset()
{
int i;
for(i=0;i<=n;i++)
father[i]=i;
return ;
}

bool cmp(Edge e1, Edge e2) // 边排序
{
return e1.w<e2.w;
}
int findfather(int x)
{
if(x!=father[x])
father[x]=findfather(father[x]);
return father[x];
}

void unionSet(int x,int y)
{
father[x]=y;
return ;
}
void krustal()
{
int i,fa,fb;
makeset();
    sort(edge,edge+edgenum,cmp);

for(i=0;i<edgenum;i++)
{
  fa=findfather(edge[i].x);
  fb=findfather(edge[i].y);

  if(fa!=fb)
  {
   unionSet(fa,fb);
   minCost+=edge[i].w;

  }
}
return ;
}

int main()
{
int i,j,k,cost;
char v1[5],v2[5];
while(scanf("%d",&n) && n)
{
  edgenum=0;
  for(i=0;i<n-1;i++)
  {
   scanf("%s%d",v1,&k);

   for(j=0;j<k;j++)
   {
    scanf("%s%d",v2,&cost);
    edge[edgenum].x=v1[0]-'A';
    edge[edgenum].y=v2[0]-'A';
    edge[edgenum++].w=cost;
   }
  }
  minCost=0;
  krustal();
  printf("%d\n",minCost);

}

return 0;
}