码迷,mamicode.com
首页 > 其他好文 > 详细

最小生成树模板

时间:2018-11-06 13:34:14      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:int   ++   clu   生成   cost   kruskal算法   turn   最小生成树   find   

Kruskal算法

伪代码

sort间接排序r[i]数组,找出权值最小的边

进行判断如果不是在一块的,那么答案加上权值两条边合并

模板

#include <bits/stdc++.h>
using namespace std;
int f[200010],r[200010];
int num[200010],to[200010],cost[200010];
int n,m;
bool cmp(int a,int b)
{
  return cost[a]<cost[b];
}
int find(int k)
{
  return f[k]==k?k:f[k]=find(f[k]);
}
int main()
{
  cin>>n>>m;
  int ans=0,sum=0;
  for(int i=0;i<m;i++)
  cin>>num[i]>>to[i]>>cost[i];
  for(int i=0;i<n;i++)
  f[i]=i;
  for(int i=0;i<m;i++)
  r[i]=i;
  sort(r,r+m,cmp);
  for(int i=0;i<m;i++)
  if(find(num[r[i]])!=find(to[r[i]]))
  {
    ans+=cost[r[i]];
    f[find(num[r[i]])]=find(to[r[i]]);
  }
  cout<<ans;
}

最小生成树模板

标签:int   ++   clu   生成   cost   kruskal算法   turn   最小生成树   find   

原文地址:https://www.cnblogs.com/baccano-acmer/p/9914566.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!