[MST] hdu 1875 畅通工程再续

Z - 畅通工程再续
Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit

Status
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!

最小生成树，这里使用的是Prim算法，
先计算距离并建图。对于大于1000米或小于10米的距离，不建立相对的距离（设置距离为INF）
然后用prim算法建立最小生成树。查找每一个节点相对的最小权值边，如果该权值距离大于1000米，直接输出”oh!”并结束，
最后输出所需要的总距离。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN = 105;
struct node
{
    double x,y;
}p[MAXN];

double cost[MAXN][MAXN];
double mincost[MAXN];
bool used[MAXN];
int T,num;

double Distance(double x1,double x2,double y1,double y2)
{
    return sqrt((x2-x1)*(x2-x1) + (y2-y1)*(y2-y1));
}
int main()
{
    cin>>T;
    while(T--)
    {
        cin>>num;

        for(int i = 0 ;i<num;i++)
            scanf("%lf%lf",&p[i].x,&p[i].y);

        memset(cost,0,sizeof(cost));

        for(int i = 0; i<num; i++)
        {
            for(int j = 0;j<num;j++)
            {
                if(100*Distance(p[i].x,p[j].x,p[i].y,p[j].y)>1000 ||
                   100*Distance(p[i].x,p[j].x,p[i].y,p[j].y)<10 )
                    cost[i][j] = cost[j][i] = INF;
                cost[i][j] = cost[j][i] = 100*Distance(p[i].x,p[j].x,p[i].y,p[j].y);
            }
        }

        for(int i = 0; i<num;i++)
        {
            mincost[i] = INF;
            used[i] = false;
        }
        mincost[0] = 0;
        double res = 0;
        while(true)
        {
            int v = -1;
            for(int u = 0 ;u<num;u++)
            {
                if(!used[u] && (v == -1 || mincost[u] < mincost[v])) v = u;
            }
            if (v == -1) break;
            used[v] = true;
            res += mincost[v];
            for(int u = 0 ;u<num;u++)
            {
                mincost[u] =min(mincost[u],cost[v][u]);
            }
        }
        if(res > 100000) cout<<"oh!"<<endl;
        else printf("%.1lf\n",res);
    }
    return 0;
}