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