最小生成树之Kruskal算法

struct Edge
{
  int u;
  int v;
  int w;
}edge[n];
bool cmp(Edge a,Edge b) //间接排序函数
{
  return a.w<b.w;
}
int find(int x)
{
  return p[x]==x?x:p[x]=find(p[x]);
}
int kruskal()
{
    for(int i=0;i<m;i++)
   {
     p[i]=i;  //初始化并查集
   }
    sort(edge,edge+n,cmp); //给边进行排序
    int sum=0,ans=0;
    for(int i=0;i<n;i++) //n为已知的总边数
    {
      int x=find(edge[i].u);
      int y=find(edge[i].v);
      if(x!=y)   //如果不在同一个集合里,则合并
      {
        ans+=edge[i].w;
        p[x]=y;
        sum++;
      }
      if(sum==m-1)
      return ans;
    }
}


阅读更多
个人分类: 最小生成树
上一篇C/C++ 基本类型数据的数值范围
下一篇C++ STL之vector
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭