AcWing基础课最小生成树模板

Prim算法（$O(n^2)$）

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