kruskal算法-Pascal

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

 const
     maxn=;
     maxe=maxn*maxn;

 type
     edge=record                            //edge记录每一条边，a,b为它所连接的两个点，len为边长
             a,b:longint;
             len:longint;
         end;

 var
     edges:array[..maxe]of edge;
     p,r:array[..maxn]of longint;
     n,e:longint;

 procedure swap(a,b:integer);
     begin
         edges[]:=edges[a];
         edges[a]:=edges[b];
         edges[b]:=edges[];
     end;

 procedure qsort(l,r:integer);                    //快排
     var
         i,j,mid,t:integer;
     begin
         mid:=edges[random(r-l+)+l].len;
         i:=l; j:=r;
         repeat
             while edges[i]<mid do inc(i);
             while edges[j]>mid do dec(j);
             if i<=j then
                 begin
                     swap(i,j);
                     inc(i);
                     dec(j);
                 end;
          until i>j;
         if l<j then qsort(l,j);
         if r>i then qsort(i,r);
     end;

 procedure init;                                                //输入
     var
         i:integer;
     begin
         assign(input,'g.in');
         reset(input);
         readln(n,e);
         for i:= to e do
             readln(edges[i].a,edges[i].b,edges[i].len);
         for i:= to n do
             p[i]:=i;
         randomize;
     end;

 function find(x:integer):integer;              //并查集：查找x的祖先
     begin
         if x<>p[x] then p[x]:=find(p[x]);
         exit(p[x]);
     end;

 procedure union(a,b:longint);                            //并查集：合并a,b所在的集合
     var
         t:integer;
     begin
         a:=find(a);
         b:=find(b);
         if r[a]>r[b] then
             begin
                 t:=a;
                 a:=b;
                 b:=t;
             end;
         if r[a]=r[b] then inc(r[a]);
         p[a]:=b;
     end;

 procedure kruskal;                                //kruskal算法
     var
         en:longint;                //记录当前为第几条边
         count:longint;        //记录当前加入了几条边
         tot:longint;            //记录当前已加入边的边权和
     begin
         count:=;
         en:=;
         tot:=;                //初始化
         while count<n- do                //当已加入的边数小于（n-）个时
             begin
                 inc(en);
                 with edges[en]do
                     begin
                         if find(a)<>find(b) then        //如果a,b不在同一集合内
                             begin
                                 union(a,b);            //合并a,b
                                 writeln(a,'--',b,':',len);
                                 inc(tot,len);
                                 inc(count);
                             end;
                     end;
             end;
     end;

 begin                //main
     init;
     qsort(,e);        //根据边长大小进行快排
     kruskal;
 end.