最小生成树

最小生成树

最小生成树对应的图是无向图

Prim

稠密图 朴素版 O(n²)

集合：当前已经在连通块当中的所有点

Dijkstra迭代n-1次 上来已经选中了一个点
Prim迭代n次 第一个点要选

典型题：铺路
最小生成树由于没有环，正边和负边都可以

不能有自环，必须先累加再更新，否则会把自环加进来

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
int g[N][N],dis[N];
bool st[N];
int n,m;
int res;
bool prim(){
    
   for(int i=0;i<n;i++){//迭代n次
       int t=-1;
       
       for(int j=1;j<=n;j++){
           if(!st[j]&&(t==-1||dis[t]>dis[j])){
               t=j;
           }
       }
       
       if(i&&dis[t]==INF)return false;//最开始没有点加入，距离一定为INF,排除该项
       
       if(i)res+=dis[t];//第一个点的距离不加入
       for(int j=1;j<=n;j++){
           if(!st[j])dis[j]=min(dis[j],g[t][j]);//与Dijksra不同这里是点到点集距离不是点到第一个点距离
       }
       
       //1.一定要先累加再更新，上面两个顺序不能反，避免负数自环
       //2.或者对输入数据进行处理，但运行速度会变慢
       
       st[t]=true;
       
   }
   return true;
}
int main(){
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof(g));
    memset(dis,0x3f,sizeof(dis));
    while(m--){
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        if(a==b)continue;//避免负数自环
        g[a][b]=g[b][a]=min(g[a][b],w);//无向图连两条边，出现重边时选择最小的
        
    }
    
    if(prim()){
        printf("%d",res);
    }else{
        puts("impossible");
    }
    return 0;
}

稀疏图 堆优化版 O(mlogn)

不常用

Kruskal O（mlogm)

时间复杂度主要花在排序上
主要用于稀疏图
不需要邻接表和邻接矩阵存图，只用把每条边存起来就行，需要用到并查集

#include <iostream>
#include <algorithm>
using namespace std;
const int N=2e5+100;
int n,m,s=1;
struct Edge{
    int a,b,w;
    bool operator <(const Edge&W)const{
        return w<W.w;
    }
}edge[N];
int p[N];
int find(int x){
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}
int res;
int kruskal(){
    for(int i=0;i<m;i++){
        int a=edge[i].a;
        int b=edge[i].b;
        int w=edge[i].w;
        if(find(a)==find(b))continue;
        p[find(a)]=find(b);
        res+=w;
        s++;
    }
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    
   for(int i=1;i<=n;i++){
       p[i]=i;
   }
    
   for(int i=0;i<m;i++){
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        edge[i]={a,b,w};
    }
    sort(edge,edge+m);
    
    int ans=kruskal();
    if(s==n)printf("%d",ans);
    else puts("impossible");
    return 0;
}