【最小生成树】Prim&Kruskal

Prim

  • 初始化w[][](a[][]),dis[],f[](vis[]); i=1~n-1中循环j=1~n找到dis[j]<min的未被标记的点k,更新min,k=j,j=1~n更新w[k][j]<dis[j]的最短距离;

  • P1546 最短网络 Agri-Net https://www.luogu.org/problemnew/show/P1546
    #include <bits/stdc++.h>
    int w[105][105],dis[105],f[105],n,s;
    const int M=2100000;
    using namespace std;
    void prim(int vi){
        for(int i=1;i<=n;i++){
            if(w[vi][i]!=0){
                dis[i]=w[vi][i];
            }
            else dis[i]=M;
        }
        f[vi]=1;
        int k,min;
        for(int i=1;i<=n-1;i++){
            min=M;
            for(int j=1;j<=n;j++){
                if(f[j]==0&&dis[j]<min){
                    min=dis[j];
                    k=j;
                }
            }
            s+=min;
            f[k]=1;
            for(int j=1;j<=n;j++){
                if(dis[j]>w[k][j]){
                    dis[j]=w[k][j];
                }
            }
        }
    }
    int main(){
        cin>>n;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                cin>>w[i][j];
            }    
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(w[i][j]==0) w[i][j]=M;
            }
        }
        prim(1);
        cout<<s<<endl;
        return 0;
    }

     

kruskal

  • sort排序;并查集;fu==fv
  • P3366 【模板】最小生成树 https://www.luogu.org/problemnew/show/P3366
    #include <bits/stdc++.h>
    using namespace std;
    struct node{
        int u,v,w;
    }e[200005];
    int fa[200005];
    int cmp(node x,node y){
        if(x.w<y.w) return 1;
        return 0;    
    };
    int getfather(int q){
        if(fa[q]!=q){
            fa[q]=getfather(fa[q]);
        }
        return fa[q];    
    }
    int main(){
        int n,m;
        cin>>n>>m;
        for(int i=1;i<=m;i++){
            int x,y,z;
            cin>>x>>y>>z;
            e[i].u=x;e[i].v=y;e[i].w=z;
        }
        sort(e+1,e+m+1,cmp);
        for(int i=1;i<=n;i++){
            fa[i]=i;
        }
        int fu,fv,s=0,ans=0;
        for(int i=1;i<=m;i++){
            fu=getfather(e[i].u);
            fv=getfather(e[i].v);
            if(fu!=fv){
                fa[fu]=fv;
                s++;
                ans+=e[i].w;
                if(s==n-1) break;
            }
        }
        cout<<ans<<endl;
        return 0;
    }

     

上一篇:【数据结构】 最小生成树(三)——prim算法


下一篇:poj2689 Prime Distance题解报告