最小生成树

【定义】

最小生成树（minimum spanning tree）是由n个顶点，n-1条边，将一个连通图连接起来，且使权值最小的结构。

【实现】

1.朴素Prim算法   O(n^2)

Prim算法总是维护最小生成树的一部分

主要用于稠密图，尤其是完全图的最小生成树求解。

 

联系：Dijkstra算法是更新到起始点的距离，Prim是更新到集合S的距离 

例题 Acwing860

【代码】

#include <bits/stdc++.h>

using namespace std;

const int N=510,INF=0x3f3f3f3f;
int g[N][N],dist[N];
bool st[N];
int n,m;

int prim(){
    memset(dist,0x3f,sizeof(dist));//初始化距离为正无穷
    int res=0;//表示集合外的点到集合的距离
    
    for(int i=0;i<n;i++){
        int t=-1;
        
        for(int j=1;j<=n;j++){
            if(!st[j] && (t==-1||dist[t]>dist[j])) 
                t=j;//用t更新距离集合最近的点
        }
        if(i && dist[t]==INF) return INF;//如果当前点不是第一个点且集合外的点与集合不连通
            
        if(i) res+=dist[t];
        for (int j = 1; j <= n; j ++ ) dist[j] = min(dist[j], g[t][j]);//attention
        st[t]=true;//更新
    }
    return res;
}
int main(){
    scanf("%d%d", &n, &m);

    memset(g, 0x3f, sizeof g);

    while (m -- ){
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        g[a][b] = g[b][a] = min(g[a][b], c);//处理重边，取最小
    }

    int t = prim();

    if (t == INF) cout<<"impossible";
    else printf("%d\n", t);

    return 0;
}

2.Kruskal算法   O(mlogm)

【分析】

1.将所有边的权重从小到大排序（快排）；

2.枚举所有边a-b，权重为w，如果a，b不连通，则将这条边加入集合当中。

（ehhhhh要用到并查集）

【代码】

#include <bits/stdc++.h>
using namespace std;
const int N=200010;
int p[N];
int n,m;
struct Edge{
    int a,b,w;
}edges[N];
bool cmp(Edge x,Edge y){
    return x.w<y.w;
}//将权重从小到大排序
int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        edges[i]={a,b,w};
    }
    sort(edges,edges+m,cmp);
    for(int i=1;i<=n;i++) p[i]=i;//初始化并查集
    int res=0,cnt=0;//分别用来存最小生成树的权重和放入集合的边数
    for(int i=0;i<m;i++){
        int a=edges[i].a,b=edges[i].b,w=edges[i].w;
        a=find(a),b=find(b);
        //如果a和b不在一个集合里就连起来
        if(a!=b){
            p[a]=b;
            res+=w;//更新权重
            cnt++;//更新边数
        }
    }
    if(cnt<n-1) cout<<"impossible";//连接n个点至少需要n-1条边，如果cnt< n-1，说明不连通
    else cout<<res;
    return 0;
}