1. 程式人生 > >USACO3.1 最短網路 Agri-Net(並查集)

USACO3.1 最短網路 Agri-Net(並查集)

最小生成樹II

Time Limit:10000MS  Memory Limit:65536K
Total Submit:301 Accepted:136 
Case Time Limit:1000MS

Description

  農民約翰被選為他們鎮的鎮長!他其中一個競選承諾就是在鎮上建立起網際網路,並連線到所有的農場。當然,他需要你的幫助。約翰已經給他的農場安排了一條高速的網路線路,他想把這條線路共享給其他農場。為了用最小的消費,他想鋪設最短的光纖去連線所有的農場。你將得到一份各農場之間連線費用的列表,你必須找出能連線所有農場並所用光纖最短的方案。每兩個農場間的距離不會超過100000

Input

第一行: 農場的個數,N(3<=N<=5000)。 
第二行..結尾: 後來的行包含了一個N*N的矩陣,表示每個農場之間的距離。理論上,他們是N行,每行由N個用空格分隔的陣列成,實際上,他們限制在80個字元,因此,某些行會緊接著另一些行。當然,對角線將會是0,因為不會有線路從第i個農場到它本身。 

Output

只有一個輸出,其中包含連線到每個農場的光纖的最小長度。

Sample Input

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

Sample Output

28
分析:按邊權從小到大排序,然後做 kruskal,用並查集判斷兩個地點是否在同一連通分量,不在就合併,由於有序所以一定是最小的。
程式碼
const
  maxn=5000*5000;
var
  a:array[0..maxn,1..3] of longint;
  p:array[0..100000] of longint;
  i,j,n,m,ans:longint;


procedure qsort(l,r:longint);
var
  i,j,k:longint;
begin
  if l>=r then exit;
  i:=l;j:=r;
  k:=a[(l+r) div 2,3];
  repeat
    while a[i,3]<k do inc(i);
    while a[j,3]>k do dec(j);
    if i<=j then
      begin
        a[0]:=a[i];a[i]:=a[j];a[j]:=a[0];
        inc(i);dec(j);
      end;
  until i>j;
  qsort(i,r);
  qsort(l,j);
end;


function find(x:longint):longint;
begin
  if (p[x]=0) then exit(x);
  p[x]:=find(p[x]);
  exit(p[x]);
end;


procedure main;
var
  i,j,k,p1,p2:longint;
begin
  for i:=1 to m do
    begin
      p1:=find(a[i,1]);
      p2:=find(a[i,2]);
      if p1<>p2 then
        begin
          p[p1]:=p2;
          inc(ans,a[i,3]);
        end;
    end;
end;


begin
  readln(n);
  m:=0;
  for i:=1 to n do
   begin
    for j:=1 to n do
      begin
        read(ans);
        inc(m);
        a[m,1]:=i;
        a[m,2]:=j;
        a[m,3]:=ans;
      end;
    readln;
   end;
  ans:=0;
  qsort(1,m);
  main;
  writeln(ans);
end.