HDU 1875 畅通工程再续(最小生成树)

HDU 1875 畅通工程再续(最小生成树)

http://acm.hdu.edu.cn/showproblem.php?pid=1875

题意:

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

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。

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

Output

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

分析:

计算出任意两点的距离,如果合法则添加对应边,之后用Kruskal模板即可.

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=100+10;
const int maxm=10000+10;

struct Edge
{
    int u,v;
    double dist;
    Edge(){}
    Edge(int u,int v,double d):u(u),v(v),dist(d){}
    bool operator<(const Edge &rhs)const
    {
        return dist<rhs.dist;
    }
};

struct Kruskal
{
    int n,m;
    Edge edges[maxm];
    int fa[maxn];
    int findset(int x){ return fa[x]==-1?x:findset(fa[x]); }

    void init(int n)
    {
        this->n=n;
        m=0;
        memset(fa,-1,sizeof(fa));
    }

    void AddEdge(int u,int v,double dist)
    {
        edges[m++]=Edge(u,v,dist);
    }

    double kruskal()
    {
        double sum=0;
        int cnt=0;
        sort(edges,edges+m);

        for(int i=0;i<m;i++)
        {
            int u=edges[i].u, v=edges[i].v;
            if(findset(u) != findset(v))
            {
                fa[findset(u)] = findset(v);
                sum += edges[i].dist;
                if(++cnt>=n-1) return sum;
            }
        }
        return -1.0;
    }
}KK;

struct Point
{
    double x,y;
}p[maxn];

double get_dist(int i,int j)//获得第i点与第j点的距离
{
    return sqrt((p[i].x-p[j].x)*(p[i].x-p[j].x)+(p[i].y-p[j].y)*(p[i].y-p[j].y));
}

int main()
{
    int T; scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        KK.init(n);
        for(int i=0;i<n;i++)
            scanf("%lf%lf",&p[i].x,&p[i].y);
        for(int i=0;i<n;i++)
        for(int j=i+1;j<n;j++)
        {
            double len=get_dist(i,j);
            if(len>=10.0 && len<=1000.0)
                KK.AddEdge(i,j,len);
        }
        double ans=KK.kruskal();
        if(ans<0 ) printf("oh!\n");
        else printf("%.1lf\n",ans*100);
    }
    return 0;
}