kruskal算法
将输入的边按照权值大小排序,可以sort也可以用优先队列,优先队列时间复杂度低一点。按照权值由小到大遍历每一条边,如果边的两个顶点不在同一个连通图中,则加入这条边,直到所有的顶点在同一个连通图中。判断是否在同一个连通图可用并查集。
注意:并查集要路径压缩,否则会超时。解题前首先判断给定的图是否是一个连通图,若不是则没有最小生成树。
#include <iostream>
#include <map>
#include <string>
#include<bits/stdc++.h>
using namespace std;
struct node{
int s;
int e;
int w;
}eage[200010];
int per[5005],n,m;
int search(int i){
if(per[i]==i)return i;
else return per[i]=search(per[i]);//路径压缩
};
int cmp(node a,node b){
return a.w<b.w;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)per[i]=i;
for(int i=0;i<m;i++){
cin>>eage[i].s>>eage[i].e>>eage[i].w;
}
//cout<<endl;
sort(eage,eage+m,cmp);
int ans=0,cnt=0;
for(int i=0;i<m;i++){
int st=search(eage[i].s),en=search(eage[i].e);
if(st==en)continue;
per[en]=st;
ans+=eage[i].w;
cnt++;
//cout<<eage[i].w<<endl;
if(cnt==n)break;
}
cout<<ans;
}