題目背景
農民約翰被選為他們鎮的鎮長!他其中一個競選承諾就是在鎮上建立起網際網路,並連線到所有的農場。當然,他需要你的幫助。
題目描述
約翰已經給他的農場安排了一條高速的網路線路,他想把這條線路共享給其他農場。為了用最小的消費,他想鋪設最短的光纖去連線所有的農場。
你將得到一份各農場之間連線費用的列表,你必須找出能連線所有農場並所用光纖最短的方案。每兩個農場間的距離不會超過100000
輸入輸出格式
輸入格式:第一行: 農場的個數,N(3<=N<=100)。
第二行..結尾: 後來的行包含了一個N*N的矩陣,表示每個農場之間的距離。理論上,他們是N行,每行由N個用空格分隔的陣列成,實際上,他們限制在80個字元,因此,某些行會緊接著另一些行。當然,對角線將會是0,因為不會有線路從第i個農場到它本身。
輸出格式:只有一個輸出,其中包含連線到每個農場的光纖的最小長度。
輸入輸出樣例
輸入樣例#1:
4 0 4 9 21 4 0 8 17 9 8 0 16 21 17 16 0
輸出樣例#1:
28
說明
題目翻譯來自NOCOW。
USACO Training Section 3.1
裸的kurskal,注意fa陣列一定要開的夠大!
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<cmath> 5 #include<algorithm> 6 using namespace std; 7 const int MAXN=4001; 8 void read(int & n) 9 { 10 char c='+';int x=0; 11 while(c<'0'||c>'9')c=getchar(); 12 while(c>='0'&&c<='9') 13 x=x*10+(c-48),c=getchar(); 14 n=x; 15 } 16 struct node 17 { 18 int u,v,w,nxt; 19 }edge[MAXN*5]; 20 int head[MAXN]; 21 int num=1; 22 void add_edge(int x,int y,int z) 23 { 24 edge[num].u=x; 25 edge[num].v=y; 26 edge[num].w=z; 27 edge[num].nxt=head[x]; 28 head[x]=num++; 29 } 30 int fa[MAXN*6]; 31 int n; 32 inline int comp(const node &a ,const node & b) 33 { 34 return a.w<b.w; 35 } 36 inline int find(int x) 37 { 38 if(fa[x]!=x) 39 return fa[x]=find(fa[x]); 40 return fa[x]; 41 } 42 inline void unionn(int x,int y) 43 { 44 int fx=find(x); 45 int fy=find(y); 46 fa[fx]=fy; 47 } 48 inline void kruskal() 49 { 50 int tot=0; 51 int ans=0; 52 sort(edge+1,edge+num,comp); 53 for(int i=1;i<=num-1;i++) 54 { 55 int now=edge[i].u;int will=edge[i].v; 56 if(find(now)!=find(will)) 57 { 58 unionn(now,will); 59 tot++; 60 ans+=edge[i].w; 61 } 62 if(tot==n-1) 63 break; 64 } 65 printf("%d",ans); 66 } 67 int main() 68 { 69 //freopen("agrinet.in","r",stdin); 70 //freopen("agrinet.out","w",stdout); 71 read(n); 72 for(int i=0;i<=n;i++) 73 fa[i]=i; 74 for(int i=1;i<=n;i++) 75 for(int j=1;j<=n;j++) 76 { 77 int x; 78 read(x); 79 if(x!=0) 80 add_edge(i,j,x); 81 } 82 kruskal(); 83 return 0; 84 }