Prim算法( O ( n 2 ) O(n^2) O(n2))
1)算法流程:
设集合S为生成树集合,T为原始集合
n次操作:
1、找到集合中T中距离距离集合S最近的一个点加入集合S中
2、利用该点更新T中其余点距离集合S的距离
#include<bits/stdc++.h>
using namespace std;
const int N=510;
int n,m;
int g[N][N];
int dist[N];
bool vis[N];
int prim(){
int res=0;//计算最小生成树的权值
for(int i=0;i<n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&(t==-1||dist[t]>dist[j])) t=j;
}
if(i&&dist[t]==0x3f3f3f3f) return 0x3f3f3f3f;//如果距离集合S的距离为INF则说明不能构成生成树
if(i) res+=dist[t];//先将选择的边计入S集合中,防止出现自环等产生错误结果
vis[t]=true;
for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);//更新到集合S的距离
}
return res;
}
int main(){
cin>>n>>m;
memset(g,0x3f,sizeof g);
memset(dist,0x3f,sizeof dist);
while(m--){
int a,b,c; cin>>a>>b>>c;
g[a][b]=g[b][a]=min(g[a][b],c);
}
int t=prim();
if(t==0x3f3f3f3f) cout<<"impossible";
else cout<<t;
return 0;
}
Kruskal算法( O ( m l o g m ) O(mlogm) O(mlogm))
1)算法流程:
将边按照权重递增排序
从小到大枚举每条边,若选取边的两端点不在一个集合内,合并集合
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10,M=2e5+10;
int f[N];
struct P{
int a,b,w;
bool operator<(const P&x)const{
return w<x.w;
}
}p[M];
int n,m;
int find(int x){
if(f[x]!=x) f[x]=find(f[x]);
return f[x];
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++) f[i]=i;
for(int i=0;i<m;i++){
int a,b,c; cin>>a>>b>>c;
p[i]={a,b,c};
}
sort(p,p+m);
int res=0,cnt=0;
for(int i=0;i<m;i++){
auto [a,b,w]=p[i];
a=find(a),b=find(b);
if(a!=b){
f[a]=b;
res+=w;
cnt++;
}
}
if(cnt<n-1) cout<<"impossible";
else cout<<res;
return 0;
}