HDU 1875 畅通工程再续【kruskal】

Problem Description

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

Output

每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.

Sample Input

2
2
10 10
20 20
3
1 1
2 2
1000 1000

Sample Output

1414.2
oh!

 

code:

   #include<stdio.h>
   #include<stdlib.h>
   #include<string.h>
   #include<math.h>
   int r[6000],u[6000],v[6000],p[102];
   double w[6000];
   int n,m;
   int cmp(const void*p1,const void*p2)
   {
    return w[*(int*)p1]>w[*(int*)p2]?1:-1;
   }
   int find(int x)
   {
    int r=x;
    while(p[r]!=r)
     r=p[r];
    int i=x;
    int j;
    while(i!=r)
    {
     j=p[i];
     p[i]=r;
     i=j;
    }
    return r;
   }
   void join(int x,int y)
   {
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
     if(fx<fy)
      p[fy]=fx;
     else
      p[fx]=fy;
    }
   }
   double kruskal()
   {
    double ans=0.0;
    int x,y,i,e;
    for(i=0;i<n;i++)
     p[i]=i;
    for(i=0;i<m;i++)
     r[i]=i;
    qsort(r,m,sizeof(r[0]),cmp);
    for(i=0;i<m;i++)
    {
     e=r[i];
     x=find(u[e]);
     y=find(v[e]);
     if(x!=y)
     {
      ans+=w[e];
      join(x,y);
     }
    }
    return ans;
   }
   int main()
   {
    int t,i,j,flag;
    int vis[100];
    double a[100][2],dis;
    scanf("%d",&t);
    while(t--)
    {
     memset(vis,0,sizeof(vis));
     m=flag=0;
     scanf("%d",&n);
     for(i=0;i<n;i++)
      scanf("%lf%lf",&a[i][0],&a[i][1]);
     for(i=0;i<n-1;i++)
      for(j=i+1;j<n;j++)
      {
dis=sqrt((a[j][0]-a[i][0])*(a[j][0]-a[i][0])+(a[j][1]-a[i][1])*(a[j][1]-a[i][1]));
       if(dis>=10&&dis<=1000)
       {
        u[m]=i;
        v[m]=j;
        w[m]=dis;
        vis[i]=1;
        vis[j]=1;
        m++;
       }
      }
     for(i=0;i<n;i++)
      if(vis[i]==0)
      {
       flag=1;
       break;
      }
     if(flag)
      printf("oh!\n");
     else
      printf("%.1lf\n",kruskal()*100);
    }

    return 0;
   }