HDU 1875 畅通工程再续(prim)

畅通工程再续

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 27578 Accepted Submission(s): 8952

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!

题解：prim

代码：

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

using namespace std;
const double INF=0x3f3f3f3f*1.0;
struct node
{
    int x;
    int y;
}n[105];

double dis[105][105];
int c;//岛屿个数
void prim()
{
    double sum=0,lowcast[105];
    memset(lowcast,0,sizeof(lowcast));
    int cnt=1;
    for(int i=0;i<c;i++)
    {
        lowcast[i] = dis[0][i];
    }
    for(int i=0;i<c-1;i++)
    {
        double minn = INF;//INF是自定义常量，表无穷大
        int k = 105;
        for(int j=0;j<c;j++)
            if(minn>lowcast[j]&&lowcast[j])
            {
                minn = lowcast[j];
                k = j;
            }
        if(k!=105)
        {
            sum+=lowcast[k];
            lowcast[k]=0;
            cnt++;
        }
        for(int j=0;j<c;j++)
            if(dis[j][k]<lowcast[j])
            {
                lowcast[j] = min(dis[j][k],lowcast[j]);
            }
    }
    if(cnt==c)
        printf("%.1lf\n",sum*100);
    else
        cout<<"oh!"<<endl;
}
int main()
{

    int t;
    cin>>t;
    while(t--)
    {
        cin>>c;
        for(int i=0;i<c;i++)
            cin>>n[i].x>>n[i].y;
        for(int i=0;i<c;i++)
            for(int j=0;j<c;j++)
            {
                if(i==j)
                {
                    dis[i][j]=0;
                    continue;
                }

                int x1=n[i].x, x2=n[j].x;
                int y1=n[i].y, y2=n[j].y;
                double l=sqrt(1.0*(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
                if(l<10.0||l>1000.0)
                    dis[j][i]=dis[i][j]=INF;
                else
                    dis[j][i]=dis[i][j]=l;
            }
            prim();
    }
    return 0;
}