HDU1875——畅通工程再续(最小生成树：Kruskal算法)

畅通工程再续

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!

题目大意：

　　　　中文，略。

解题思路：

　　　　简单的最小生成树，用Kruskal算法过的。

　　　　这个题的顶点是用(x,y)存的，可以用结构体存。

　　　　通过判断是否会有N-1条边加入树来判断是否可以生成最小树。

　　　　PS：sqrt()里面要是INT型等整型，需要强制转换，因为math.h中有 double sqrt(double),float sqrt(flaot),long double sqrt(long double)三个函数。HDU会报编译错误。

Code：

#include<stdio.h>
#include<string>
#include<vector>
#include<algorithm>
#include<cmath>
#include<iostream>
#define MAXN 5000
using namespace std;
struct point
{
    int x,y;
    int father;
} P[120];
struct edge
{
    int begin;
    int end;
    double dis;
} T[MAXN+10];
void init()
{
    for (int i=1; i<=MAXN; i++)
        P[i].father=i;
}
int find(int x)
{
    if (P[x].father!=x)
        P[x].father=find(P[x].father);
    return P[x].father;
}
void join(int x,int y)
{
    int fx=find(x),fy=find(y);
    if (fx!=fy)
        P[fx].father=fy;
}
bool cmp(struct edge a,struct edge b)
{
    return a.dis<b.dis;
}
double dis2(int x1,int y1,int x2,int y2)
{
    return sqrt((double)((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)));
}
int main()
{
    int S;
    cin>>S;
    while (S--)
    {
        init();
        int C;
        cin>>C;
        for (int i=1; i<=C; i++)
            cin>>P[i].x>>P[i].y;
        int k=1;
        bool ok=0;
        for (int i=1; i<=C; i++)
            for (int j=1; j<i; j++)
            {
                T[k].dis=dis2(P[i].x,P[i].y,P[j].x,P[j].y);
                T[k].begin=i;
                T[k].end=j;
                k++;
            }
        sort(T+1,T+k,cmp);
        int sum=0;
        double cnt=0;

        for (int i=1; i<k; i++)
        {
            if (find(T[i].begin)!=find(T[i].end)&&
                (10.000000<=T[i].dis&&T[i].dis<=1000.000001))
            {
                sum++;
                cnt+=T[i].dis;
                join(T[i].begin,T[i].end);
            }
            if (sum==C-1)
            {
                ok=1;
                break;
            }
        }
        if (ok) printf("%.1lf
",cnt*100);
        else printf("oh!
");
    }
    return 0;
}