最短网络Agri-Net

问题描述】

　　农民约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了用最小的消费，他想铺设最短的光纤去连接所有的农场。你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000。

【输入格式

【输出格式】

只有一个输出，其中包含连接到每个农场的光纤的最小长度。

【输入样例】agrinet.in

4

0  4  9  21

4  0  8  17

9  8  0  16

21 17 16  0

【输出样例】agrinet.out

　　28

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 int num=1;
 7 struct node
 8 {
 9     int u;
10     int v;
11     int w;
12     int next;
13 }edge[10001];
14 int head[10001];
15 int f[10001];
16 int tot=0;
17 int cmp(const node &a,const node &b)
18 {
19     if(a.w<b.w)return 1;
20     else return 0;
21 }
22 int find(int x)
23 {
24     if(f[x]!=x)
25     f[x]=find(f[x]);
26     return f[x];
27 }
28 void unionn(int x,int y)
29 {
30     int fx=find(x);
31     int fy=find(y);
32     if(fx!=fy)
33     f[fx]=fy;
34 }
35 int main()
36 {
37     int n;
38     scanf("%d",&n);
39     for(int i=0;i<=n;i++)head[i]=-1;
40     for(int i=0;i<=n;i++)f[i]=i;
41     for(int i=1;i<=n;i++)
42     {
43         for(int j=1;j<=n;j++)
44         {
45             int kk;
46             scanf("%d",&kk);
47             if(kk!=0)
48             {
49                 edge[num].w=kk;
50                 edge[num].u=i;
51                 edge[num].v=j;
52                 edge[num].next=head[i];
53                 num++;
54             }
55         }
56     }
57     int k=0;
58     sort(edge+1,edge+num,cmp);
59     for(int i=1;i<=num-1;i++)
60     {
61         if(find(edge[i].u)!=find(edge[i].v))
62         {
63             unionn(edge[i].u,edge[i].v);
64             tot=tot+edge[i].w;
65             k++;
66         }
67         if(k==n-1)break;
68     }
69     printf("%d",tot);
70     return 0;
71 }