说了最小生成树,那么就用经典的Prim或者Kruskal,不过Prim实现代码有点多,这里用Kruskal举例。
注意事项
1.Kruskal是用来找最小生成树的.
根据树的定义可以知道,树是无向图,所以Kruskal只能用在无向图里面
2.所给的图必须是1个无向图,不然要依次分开去Kruskal.
3.如果所给的1个无向图中找不出n-1个边来建树,那就是不存在最小生成树
如何证明Kruskal在1个无向图中必然能找出(或者没有)最小生成树:登录 - 洛谷
并查集可以把集合并在一起或者查询是否在同一个集合,所以可以用并查集来判断连通性
Kruskal:
pre 对边的权值大小按照升序排序
work 从小到大开始选边
如果两个点不连通,那就加入最小生成树并且连通两点
连通那就判断下一条边
如果已经建完树了(有n-1条边了),那就完成建树
通常来说无向边建图需要开两倍边数组,Prim需要从每个点出去走一遍所以要开两倍。
但是Kruskal只判断u,v两点是否连通,他不在乎边是否有方向,都当作无向边来用。
也就是说对于Kruskal来说,输入1 2 3和2 1 3是一样的。
AC代码
#include <bits/stdc++.h>
using namespace std;
const int MAXM=2e5+1;
const int MAXN=5e3+1;
struct Edge{
int u,v,w;
}e[MAXM];
int n,m,f[MAXN],cnt,ans;
//并查集,定义f[i]=j为节点i在集合j中
int find(int x){//找出x所在的集合+路径压缩
if(f[x]==x)return x;
else return f[x]=find(f[x]);
}
inline void unit(int x,int y){//合并集合x1,x2
int f1= find(x),f2= find(y);
f[f1]=f2;
}
inline void initf(){//初始化并查集
for(int i=1;i<=n;++i)
f[i]=i;
}
bool cmp(Edge x,Edge y){
return x.w<y.w;
}
inline void kruskal(){
for(int i=1;i<=m;++i){
int u=e[i].u,v=e[i].v;
if(find(u)== find(v))continue;//如果已经在同一个集合
else{
cnt++;//加入最小生成树
ans+=e[i].w;
unit(u,v);//合并u,v
if(cnt==n-1){//如果已经有n-1条
cout<<ans;
return;
}
}
}
//边都判断完了都找不出最小生成树
cout<<"orz";
}
int main(){
ios::sync_with_stdio(false);
cin.tie(nullptr),cout.tie(nullptr);
cin>>n>>m;
for(int i=1;i<=m;++i){
cin>>e[i].u>>e[i].v>>e[i].w;
}
initf();
sort(e+1,e+1+m,cmp);//按边的权值进行升序
kruskal();
return 0;
}
例题:口袋的天空 - 洛谷