最小生成树模板

Posted 2021-01-17 baccano-acmer

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;
}