算法总结之最小生成树

最小生成树的作用:

有很多点，点点之间有很多边，边有边权，我们要选择一些边，将所有点互相联通，构成一颗树，即为最小生成树

模板题：http://codevs.cn/problem/1231/

算法：

prim

基本流程：开始从所有点 中任取一点，（通常取1号），然后找其他所有点离已经取得点构成的连通块最近的点，将其加入连通块，并将总代价加入这个点与联通块相连的边的边权，如此反复点数-1次即可。（找出离连通块最近的点可以使用堆优化）

代码：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66

#include <cstdio> #include <cmath> #include <cstring> #include <algorithm> #include <queue> using  namespace  std; struct  lx {      long  long  dis;      int  x; }; const  bool   operator < ( const  lx x, const  lx y) {      return  x.dis > y.dis; } priority_queue <lx> dl; int  head[110000],nxt[220000],to[220000],n,m; int  qd,a,b,s,sum; long  long  tot,w[220000]; bool  ins[110000]; void  cr( int  x, int  y, int  s) {      sum++;      nxt[sum] = head[x];      head[x] = sum;      to[sum] = y;      w[sum] = s; } int  main() {      scanf ( "%d%d" ,&n,&m);      for  ( int  i = 1;i <= m;i++)      {          scanf ( "%d%d%d" ,&a,&b,&s);          cr(a,b,s);          cr(b,a,s);      }      ins[1] =  true ;      qd = 1;      for  ( int  i = 1;i < n;i++)      {          long  long  minn = 999999999999999;          for  ( int  tp = head[qd];tp;tp = nxt[tp])          {              lx ttp;              ttp.x = to[tp];              ttp.dis = w[tp];              dl.push(ttp);          }          bool  pc =  true ;          while  (pc)          {              lx ptp = dl.top();              dl.pop();              if  (!ins[ptp.x])              {                  qd = ptp.x;                  pc =  false ;                  tot += ptp.dis;                  ins[ptp.x] =  true ;              }          }      }      printf ( "%lld" ,tot);      return  0; }

kruskal

基本流程：

把所有边按照边权从小到大排序，从小到大扫，如当前边的两个端点，不在同一个并查集集合中，就将他们加入同一个并查集集合，并加入这条边。若两点在同一集合则忽略这条边。

代码：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

#include <cstdio> #include <cmath> #include <cstring> #include <algorithm> using  namespace  std; int  n,m,fa[110000],siz[110000]; long  long  tot; struct  xn {      int  x,y,w; } sz[110000]; const  bool  operator <( const  xn &a, const  xn &b) {      return  a.w < b.w; } int  getfa( int  x) {      if  (fa[x] == x)  return  x;      return  fa[x] = getfa(fa[x]); } void  merge( int  x, int  y) {      int  fa1 = getfa(x),fa2 = getfa(y);      if  (siz[fa1] < siz[fa2])      {          fa[fa1] = fa2;          siz[fa2] += siz[fa1];      } else      {          fa[fa2] = fa1;          siz[fa1] += siz[fa2];      } } int  main() {      scanf ( "%d%d" ,&n,&m);      for  ( int  i = 1;i <= n;i++)      {          fa[i] = i;          siz[i] =1;      }      for  ( int  i = 1;i <= m;i++)      {          scanf ( "%d%d%d" ,&sz[i].x,&sz[i].y,&sz[i].w);      }      sort(sz+1,sz+m+1);      for  ( int  i = 1;i <= m;i++)      {          if  (getfa(sz[i].x) != getfa(sz[i].y))          {              merge(sz[i].x,sz[i].y);              tot += ( long  long )sz[i].w;          }      }      printf ( "%lld\n" ,tot);      return  0; }