【最小生成树】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;
  }

   

转载于:https://www.cnblogs.com/songjian-jiansong/p/10312574.html