hdu 1875 畅通工程再续 最小生成树

好久不刷题了，先练练手，hdu1875

畅通工程再续

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 13532    Accepted Submission(s): 4156

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!
  
  

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cmath>
using namespace std;
const int N = 10010;
struct P
{
    int from, to;
    double value;
    P(int a, int b, double c):from(a), to(b), value(c){};
};
bool cmp(P a, P b)
{
    return a.value < b.value;
}
int father[N], son[N];
double sum;
void init(int n)
{
    sum = 0;
    for(int i = 0; i < n; i++)
    {
        father[i] = i;
        son[i] = 1;
    }
}
int findf(int x)
{
    return (x == father[x]) ? x : (father[x] = findf(father[x]));
}
bool unoin(int p, int q)
{
    int fp = findf(p);
    int fq = findf(q);
    if(fp == fq)return 0;
    if(son[fp] > son[fq])
    {
        father[fq] = fp;
        son[fp] += son[fq];
    }
    else
    {
        father[fp] = fq;
        son[fq] += son[fp];
    }
    return 1;
}
double dist(int x1, int y1, int x2, int y2)
{
    return sqrt((x2 - x1) * (x2 - x1) + (y2 - y1) * (y2 - y1));
}
int main()
{
    int n;
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &n);
        vector<P> p;
        int x[110], y[110];
        for(int i = 0; i< n; i++)
        {
            scanf("%d%d", &x[i], &y[i]);
        }
        for(int i = 0; i < n; i++)
            for(int j = i + 1; j < n; j++)
            {
                double t = dist(x[i], y[i], x[j], y[j]);
                if(t > 1000 || t < 10)continue;
                p.push_back(P(i, j, t));
            }
        sort(p.begin(), p.end(),cmp);
        init(n);
        int cnt = 0;
        for(int i = 0; i < p.size(); i++)
        {
            if(unoin(p[i].from, p[i].to))
            {
                sum += p[i].value * 100;
                cnt++;
            }
        }
        if(cnt >= n - 1)
        {
            printf("%.1lf\n", sum);
        }
        else printf("oh!\n");
    }
    return 0;
}