HDU-1875 畅通工程再续

传送门：HDU - 1875

题目描述

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

输入描述

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

输出描述

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

思路分析

思路和上一题差不多，多一步计算任意两点之间的距离，再套用最小生成树模板就行了，这里用的是Kruskal算法

样例输入

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

样例输出

1414.2
oh!

AC代码

#include<bits/stdc++.h>

using namespace std;

#define ll long long
#define ull unsigned long long
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);

const int maxn=1e6+10;
const ll mod=1e9+7;
const int INF = 0x3f3f3f3f;

int n,k;
int f[10000];

struct node{
    int a,b;
    double d;
}s[10000];

struct point{
    int x,y;
}a[10000];

int cmp(node a,node b){
    return a.d<b.d;
}

int find(int x){
    if(f[x]==x)
        return x;
    else
        return f[x]=find(f[x]);
}

int unite(int a,int b){
    int t1,t2;
    t1=find(a);
    t2=find(b);
    if(t1!=t2){
        f[t2]=t1;
        return 1;
    }
    return 0;
}

double dis(point a,point b){
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}

void sta(){
    memset(s,0,sizeof(s));
    memset(a,0,sizeof(a));
    for(int i=0;i<=n;i++)
        f[i]=i;
}

void kruskal(){
    int cnt=0;
    double ans=0;
    for(int i=0;i<k;i++){
        if(unite(s[i].a,s[i].b)){
            cnt++;
            ans+=s[i].d;
        }
        if(cnt==n-1)
            break;
    }
    if(cnt<n-1)
        printf("oh!\n");
    else
        printf("%.1lf\n",ans*100);
}



int main() {
    int t;
    cin>>t;
    while(t--){
        cin>>n;
        sta();
        for(int i=0;i<n;i++){
            cin>>a[i].x>>a[i].y;
        }
        k=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                double u;
                u=dis(a[i],a[j]);
                if(u<10.0||u>1000.0)
                    continue;
                s[k].a=i;
                s[k].b=j;
                s[k++].d=u;
            }
        }
        sort(s,s+k,cmp);
        kruskal();
    }
    
    return 0;
}