HDU - 1875 畅通工程再续（最小生成树 prim算法）

畅通工程再续

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

Total Submission(s): 32814    Accepted Submission(s): 10789

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!

Author

8600

Code

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define N 110
#define INF 0x3f3f3f3f

using namespace std;

double map[N][N];
double dis[N];
int vis[N];

struct node
{
    double x, y;
} s[N];

double getDist(int i, int j)
{
    double t = sqrt( (s[i].x - s[j].x) * (s[i].x - s[j].x) + (s[i].y - s[j].y) * (s[i].y - s[j].y) );
    if(t > 1000 || t < 10)
        return INF;
    return t;
}

void prim(int n)
{
    int i,j,pos;
    double min, ans = 0;
    for(i=0; i<n; i++)
    {
        dis[i] = map[1][i];
        vis[i] = 0;
    }
    vis[1] = 1;
    for(i=0; i<n; i++)
    {
        min = INF * 1.0;
        for(j=0; j<n; j++)
        {
            if(!vis[j] && dis[j] < min)
            {
                min = dis[j];
                pos = j;
            }
        }
        if(min == INF * 1.0)
            break;
        ans += min;
        vis[pos] = 1;
        for(j=0; j<n; j++)
        {
            if(!vis[j] && dis[j] > map[pos][j])
            {
                dis[j] = map[pos][j];
            }
        }

    }
    if(ans == 0)
        printf("oh!\n");
    else
        printf("%.1lf\n",ans*100);
}

int main()
{
    int t,i,j,n;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=0; i<n; i++)
        {
            scanf("%lf %lf",&s[i].x, &s[i].y);
        }
        for(i=0; i<n; i++)
        {
            for(j=0; j<n; j++)
            {
                map[i][j] = getDist(i,j);
            }
        }
        prim(n);
    }
    return 0;
}

反思：

最小生成树模板题，注意两岛距离小于10或大于1000不建桥。