HDU - 1875_畅通工程再续

畅通工程再续

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

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!

题解：用kruskal求最小生成树，注意判断边权，然后在并茶几那里出bug了，wa了好几发。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>

using namespace std;

const int maxn = 1050;

struct point
{
    int x,y;
}p[105];

struct edge
{
    int v,u;
    double w;
}s[10050];

int num,head[maxn],n;

void add(int u,int v,double w)
{
   // cout<<u<<' '<<v<<' '<<w<<endl;
    s[num].u = u;
    s[num].v = v;
    s[num].w = w;
    num++;
}

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

int Find(int x)
{
    return x==head[x]?x:head[x] = Find(head[x]);
}

double kruskal()
{
    double w,sum = 0;
    int i,v,u,ans = 0;
    for(i=0;i<num;i++)
    {
        u = s[i].u;
        v = s[i].v;
        w = s[i].w;
        //cout<<"WTF"<<endl;
        int a = Find(u);
        int b = Find(v);
        if(a!=b&&w>=10&&w<=1000)
        {
            //cout<<w<<endl;
            head[b] = a;
            sum += w;
        }
    }
    for(i=0;i<n;i++)
    {
        if(head[i]==i)
            ans ++;
    }
    if(ans==1)
        return sum;
    return -1;
}

void init(int n)
{
    for(int i=0;i<n;i++)
        head[i] = i;
}

bool cmp(edge a,edge b)
{
    return a.w<b.w;
}

int main()
{
    int i,t,j;
    scanf("%d",&t);
    while(t--)
    {
        num = 0;
        scanf("%d",&n);
        init(n);
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&p[i].x,&p[i].y);
            for(j=0;j<i;j++)
            {
                add(i,j,dis(p[i],p[j]));
            }
        }
        if(n==1||n==0)
        {
            printf("0\n");
            continue;
        }
        sort(s,s+num,cmp);
        double a = kruskal();
        if(a==-1)
            printf("oh!\n");
        else
            printf("%.1f\n",a*100);
    }
    return 0;
}