USACO3.1 最短網路 Agri-Net(並查集)
阿新 • • 發佈:2019-02-15
最小生成樹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.