畅通工程再续 HDU - 1875（克鲁斯卡尔）

畅通工程再续

  HDU - 1875

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

题意：已知n个坐标，距离超过1000或者不足10的路无法相连，求MST，若无法连通所有点，则输出oh！

思路：MST。生成的所有边中去掉不符合规则的边，选取n-1条边，若够n-1条边，则说明可以连通，否则无法连通

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<utility>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#define maxn 505
#define INF 0x3f3f3f3f
#define LL long long
#define ULL unsigned long long
#define mod 1000000007
#define P pair<int,int>
using namespace std;

int n,m,k;
int par[maxn];
int ranknum[maxn];

struct Edge{
    int s;
    int e;
    double len;
}E[500005];

struct Point{
    int x,y;
}Po[maxn];

bool cmp(Edge x,Edge y){
    return x.len<y.len;
}
//并查集
void init(int n)
{
    for(int i=1;i<=n;++i){
        par[i] = i;
        ranknum[i] = 0;
    }
}
int findpar(int x)
{
    if(par[x]==x) return x;
    return par[x] = findpar(par[x]);
}
void unite(int x,int y)
{
    int fx = findpar(x);
    int fy = findpar(y);
    if(fx == fy) return ;
    if(ranknum[fx]<ranknum[fy]){
        par[fx] = fy;
    }
    else{
        par[fy] = fx;
        if(ranknum[fx]==ranknum[fy])
            ranknum[fx]++;
    }
}
double find_dist(int i,int j)
{
    return fabs(sqrt(1.0*(Po[i].x-Po[j].x)*(Po[i].x-Po[j].x)+1.0*(Po[i].y-Po[j].y)*(Po[i].y-Po[j].y)));
}
 double kruskal()
 {
    int cnt = 0;
    double sum=0.0;
    for(int i=1;i<=k;++i){
        if(cnt == n-1) break;
        if(findpar(E[i].s)==findpar(E[i].e)||E[i].len<10||E[i].len>1000) continue;
        unite(E[i].s,E[i].e);
        sum += E[i].len;
        ++cnt;
    }
    if(cnt<n-1) return -1;
    return sum;
 }
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        k = 0;
        for(int i=1;i<=n;++i){
            scanf("%d %d",&Po[i].x,&Po[i].y);
        }
        for(int i=1;i<=n;++i){
            for(int j=i+1;j<=n;++j){
                E[++k].s = i;
                E[k].e = j;
                E[k].len = find_dist(i,j);
            }
        }
    /*    for(int i=1;i<=k;++i){
            printf("%d %d %.5lf\n",E[i].s,E[i].e,E[i].len);
        }*/
        init(n);
        sort(E+1,E+k+1,cmp);
        double ans = kruskal();
        if(ans == -1) printf("oh!\n");
        else{
            ans *= 100;
            printf("%.1lf\n",ans);
        }
    }
    return 0;
}