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

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

题目：

畅通工程再续

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

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!

思路：

直接套模版。

代码：

#include"cstdio"
#include"iostream"
#include"string.h"
#include"math.h"
#include"stdlib.h"

using namespace std;

const double INF = 12345678.000000;
const int N = 105;
int n;
double dis[N];
int x[N];
int y[N];
double maze[N][N];
bool vis[N];

void init()
{
    for(int i=0; i<N; i++)
    {
        for(int j=0; j<N; j++)
        {
            maze[i][j]=INF;
        }
        dis[i]=INF;
    }
    memset(vis,false,sizeof(vis));
}

int prim()
{
    double ans=0;
    dis[1]=0;
    for(int i=1; i<=n; i++)
    {
        int minx=INF;
        double minn=INF;
        for(int j=1; j<=n; j++)
        {
            if(vis[j]==false&&dis[j]<=minn)
            {
                minn=dis[j];
                minx=j;
            }
        }
        vis[minx]=true;
        //同样，在没有跑完之前距离起点最短的点的距离为INF，那么就构不成最小生成树
        if(minn+10>INF)   return -1;
        ans=ans+minn;
        for(int j=1; j<=n; j++)
        {
            if(vis[j]==false&&dis[j]>maze[minx][j])
                dis[j]=maze[minx][j];
        }
    }
    ans=ans*100;
    printf("%.1f\n",ans);
    return 0;
}


int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            scanf("%d %d",&x[i],&y[i]);
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=i+1; j<=n; j++)
            {
                int a=abs(x[i]-x[j]);
                int b=abs(y[i]-y[j]);
                double len=sqrt(a*a+b*b);
                //对于不满足要求的边，我们不加进去
                if(len<10.00||len>1000.00)
                    continue;
                maze[i][j]=len;
                maze[j][i]=len;
            }
        }
        if(prim())  printf("oh!\n");
    }
    return 0;
}