HDOJ1875-----最小生成树

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组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!
  
  

题意很好理解，直接代码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
int pre[102];
struct node{
    int x, y;
    double z;
} cnt[10002];
node nu[102];
bool cmp(node a, node b){
    return a.z < b.z;
}
int find(int x){
    return x == pre[x] ? x : (pre[x] = find(pre[x]));
}
bool merge(int x, int y){
    int fx = find(x), fy = find(y);
    if(fx != fy){
        pre[fy] = fx;
        return true;
    }
    return false;
}
int main(){
    int m, n, num, ans;
    double sum;
    scanf("%d", &m);
    while(m--){
        scanf("%d", &n);
        for(int i = 0; i < n; i++){
            scanf("%d%d", &nu[i].x, &nu[i].y);//先用一组结构体保存坐标
        }
        num = sum = ans = 0;
        for(int i = 0; i < n - 1; i++){//将第一组坐标设为第一座小岛，依次遍历所有岛的距离
            for(int k = i + 1; k < n; k++){
                cnt[num].x = i;
                cnt[num].y = k;
                cnt[num++].z = sqrt((nu[i].x - nu[k].x) * (nu[i].x - nu[k].x) + (nu[i].y - nu[k].y) * (nu[i].y - nu[k].y));
            }
        }
        sort(cnt, cnt + num, cmp);//克鲁斯卡尔
        for(int i = 0; i < 102; i++){
            pre[i] = i;
        }
        for(int i = 0; i < num; i++){
            if(cnt[i].z >= 10 && cnt[i].z <= 1000 && merge(cnt[i].x, cnt[i].y)){//先判断距离，再运算并查集，ans记录一共多少条路
                sum += cnt[i].z;
                ans++;
            }
        }
        printf(ans != n - 1 ? "oh!\n" : "%.1lf\n", sum * 100);//n个小岛应该对应n - 1条路
    }
    return 0;
}