HDU-1875-畅通工程再续（最小生成树 普利姆）

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

Status

Practice

HDU 1875
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<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<math.h>
using namespace std;
//最小生成树
//普利姆算法
const int maxn=105;
const int INF=99999999;//正无穷
double map[maxn][maxn];//邻接矩阵存图
double dis[maxn];//记录各点到生成树的最短距离
int vis[maxn];//添加到生成树的点标记为1；
int T;//数据组数
int C;//小岛数量
struct node
{
    int x;//第i个小岛的坐标
    int y;
} num[maxn];
void init()//初始化邻接矩阵并存图
{
    for(int i=0; i<=C; i++)//初始化邻接矩阵
        for(int j=0; j<=C; j++)
            i==j?map[i][j]=0:map[i][j]=INF;
    scanf("%d%d",&num[1].x,&num[1].y);
    for(int i=2; i<=C; i++) //存图
    {
        scanf("%d%d",&num[i].x,&num[i].y);
        for(int j=1; j<i; j++)
        {
            double flag=sqrt((num[i].x-num[j].x)*(num[i].x-num[j].x)+(num[i].y-num[j].y)*(num[i].y-num[j].y));
            if(flag>=10&&flag<=1000)
            {
                map[i][j]=flag;//i到j距离为flag
                map[j][i]=flag;//无向图
            }
        }
    }
    memset(vis,0,sizeof(vis));//初始化生成树集合为空
    vis[1]=1;//1添加入生成树
    for(int i=1; i<=C; i++) //初始化dis数组
        dis[i]=map[1][i];
}
void Prim()//普利姆算法
{
    int count=0;//记录生成树边的数量
    double min_cost=0;//最小花费
    while(1)//有结束条件
    {
        double minn=INF;//记录松弛边
        int point_minn;//记录松弛点的下标
        for(int i=1; i<=C; i++)
            if(vis[i]==0&&dis[i]<minn)//i不在生成树内,且可以被松弛
            {
                minn=dis[i];
                point_minn=i;
            }
        if(minn==INF)//如果松弛失败
        {
            printf("oh!\n");
            break;
        }
        vis[point_minn]=1;//标记为加入生成树
        count++;//更新生成树边的数量
        min_cost+=dis[point_minn]*100;
        for(int k=1; k<=C; k++)
            if(vis[k]==0&&dis[k]>map[point_minn][k])
                dis[k]=map[point_minn][k];//迪杰斯特拉松弛
        if(count==C-1)
        {
            printf("%.1lf\n",min_cost);
            break;
        }
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&C);
        init();
        Prim();
    }
    return 0;
}