kruskal算法-Pascal

马上就快要考试了，然而突然发现自己图论已经废了，于是再都打一遍练练手。。。。。。

  1 const
  2     maxn=100;
  3     maxe=maxn*maxn;
  4
  5 type
  6     edge=record                            //edge记录每一条边，a,b为它所连接的两个点，len为边长
  7             a,b:longint;
  8             len:longint;
  9         end;
 10
 11 var
 12     edges:array[0..maxe]of edge;
 13     p,r:array[0..maxn]of longint;
 14     n,e:longint;
 15
 16 procedure swap(a,b:integer);
 17     begin
 18         edges[0]:=edges[a];
 19         edges[a]:=edges[b];
 20         edges[b]:=edges[0];
 21     end;
 22
 23 procedure qsort(l,r:integer);                    //快排
 24     var
 25         i,j,mid,t:integer;
 26     begin
 27         mid:=edges[random(r-l+1)+l].len;
 28         i:=l; j:=r;
 29         repeat
 30             while edges[i]<mid do inc(i);
 31             while edges[j]>mid do dec(j);
 32             if i<=j then
 33                 begin
 34                     swap(i,j);
 35                     inc(i);
 36                     dec(j);
 37                 end;
 38          until i>j;
 39         if l<j then qsort(l,j);
 40         if r>i then qsort(i,r);
 41     end;
 42
 43 procedure init;                                                //输入
 44     var
 45         i:integer;
 46     begin
 47         assign(input,‘g.in‘);
 48         reset(input);
 49         readln(n,e);
 50         for i:=1 to e do
 51             readln(edges[i].a,edges[i].b,edges[i].len);
 52         for i:=1 to n do
 53             p[i]:=i;
 54         randomize;
 55     end;
 56
 57 function find(x:integer):integer;              //并查集：查找x的祖先
 58     begin
 59         if x<>p[x] then p[x]:=find(p[x]);
 60         exit(p[x]);
 61     end;
 62
 63 procedure union(a,b:longint);                            //并查集：合并a,b所在的集合
 64     var
 65         t:integer;
 66     begin
 67         a:=find(a);
 68         b:=find(b);
 69         if r[a]>r[b] then
 70             begin
 71                 t:=a;
 72                 a:=b;
 73                 b:=t;
 74             end;
 75         if r[a]=r[b] then inc(r[a]);
 76         p[a]:=b;
 77     end;
 78
 79 procedure kruskal;                                //kruskal算法
 80     var
 81         en:longint;                //记录当前为第几条边
 82         count:longint;        //记录当前加入了几条边
 83         tot:longint;            //记录当前已加入边的边权和
 84     begin
 85         count:=0;
 86         en:=0;
 87         tot:=0;                //初始化
 88         while count<n-1 do                //当已加入的边数小于（n-1）个时
 89             begin
 90                 inc(en);
 91                 with edges[en]do
 92                     begin
 93                         if find(a)<>find(b) then        //如果a,b不在同一集合内
 94                             begin
 95                                 union(a,b);            //合并a,b
 96                                 writeln(a,‘--‘,b,‘:‘,len);
 97                                 inc(tot,len);
 98                                 inc(count);
 99                             end;
100                     end;
101             end;
102     end;
103
104 begin                //main
105     init;
106     qsort(1,e);        //根据边长大小进行快排
107     kruskal;
108 end.