hdu 1875 畅通工程再续(最小生成树 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!

 

 

#include<stdio.h>

#include<string.h>

#include<math.h>

#include<iostream>

#include<algorithm>

#include<queue>

#include<stack>

#define mem(a,b) memset(a,b,sizeof(a))

#define ll __int64

#define MAXN 1000

#define INF 0x7ffffff

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

using namespace std;

struct Point {

   double x,y;

};

struct Road{

   int f,t;

   double w;

};

Point p[250];

Road  r[50000];

int fat[250];

int sum,cnt,n;

double ans;

int cmp(Road a,Road b)

{

    return a.w<b.w;

}

int find(int a)

{

    //return a==fat[a]?a:find(fat[a]);

    if(a!=fat[a])

    {

        fat[a]=find(fat[a]);

    }

    return fat[a];

}

void Kruskal()

{

    for(int i=1;i<=n;i++) fat[i]=i;



    for(int i=1;i<cnt;i++)

    {

        int x=find(r[i].f);

        int y=find(r[i].t);

        if(x!=y&&r[i].w>=10&&r[i].w<=1000)

        {

            sum++;

            ans+=r[i].w;

            fat[y]=x;

        }

    }

    if(sum==n)

        {

            ans*=100;

            printf("%.1f\n",ans);

        }

        else

            cout<<"oh!\n";

}

int main()

{

    int i,j,t;

    cin>>t;

    while(t--)

    {

        scanf("%d",&n);

        sum=1;ans=0;

        for(i=1;i<=n;i++)

        {

            scanf("%lf%lf",&p[i].x,&p[i].y);

        }

        cnt=1;

        for(i=1;i<n;i++)

        {

            for(j=i+1;j<=n;j++)

            {

                r[cnt].f=i;

                r[cnt].t=j;

                r[cnt].w=sqrt(fabs(pow(p[i].x-p[j].x,2)+pow(p[i].y-p[j].y,2)));

                cnt++;                

            }

        }

        //cout<<cnt<<endl;

        sort(r+1,r+cnt,cmp);

        Kruskal();

    }

    return 0;

}