NYOJ 1875 畅通工程再续 （无节点间距离求最小生成树）

 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<cmath>
 4 using namespace std;
 5 int fa[10000];
 6 
 7 struct stu
 8 {
 9     int from,to;    
10     double al;
11 }st[100000];
12 
13 bool cmp(stu a,stu b)
14 {
15     return a.al < b.al;
16 }
17 
18 int find(int a)                                //搜索父节点函数 
19 {
20     int r=a;
21     while(r != fa[r])
22     {
23         r=fa[r];
24     }
25     return r;
26 }
27 
28 int main()
29 {
30     int t;
31     scanf("%d",&t);
32     while(t--)
33     {
34         int num,c,a[10000],b[10000],i,j;
35         num=-1;
36         scanf("%d",&c);
37         
38         for(i = 1 ; i<= c ; i++)
39         {
40             fa[i]=i;
41         }
42         
43         for(i = 1 ; i <= c ;i++)
44         {
45             scanf("%d %d",&a[i],&b[i]);            //数组a[]和数组b[]分别存放第i个岛的x,y坐标 
46         }
47         
48         for(i = 1 ; i < c ; i++)
49         {
50             for(j = i+1 ; j <= c ; j++)
51             {
52                 num++;
53                 st[num].from=i;                    //结构体存放第i个岛和第j个岛于它们之间的距离 
54                 st[num].to=j;
55                 st[num].al=sqrt((a[i]-a[j])*(a[i]-a[j])+(b[i]-b[j])*(b[i]-b[j]));
56             }
57         }
58         
59         sort(st,st+num+1,cmp);                    //对岛之间的距离从小到大排序 
60         int k=0;
61         double sum=0;
62         
63         for(i = 0 ; i <= num ; i++)
64         {
65             if(k == c-1)                        //连接c个岛只须c-1个边 
66             {
67                 break;
68             }
69             
70             if(find(st[i].from) != find(st[i].to) && st[i].al >=10 && st[i].al <= 1000)
71             {
72                 fa[find(st[i].from)]=find(st[i].to);
73                 sum+=st[i].al;
74                 k++;
75             }
76         }
77         
78         if(k == c-1)                                //若找不到c-1条边则不能将岛都连起来 
79             printf("%.1lf\n",sum*100);
80         else
81             printf("oh!\n");
82     }
83  }