畅通工程再续hdu-1875(prim算法)

题目地址:[传送门(http://acm.hdu.edu.cn/showproblem.php?pid=1875 “传送门”)

本题使用的是prim算法,用于求最小生成树。

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!

AC代码如下:(代码中有注释。)
技术分享

#include<cmath>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=111;
const int inf=111111;
struct LOAD
{
    int x,y;
} load[maxn];
double mincost[maxn],road[maxn][maxn];
int T,n,used[maxn],ok;
double dis(int i,int j)//用来求i到j岛的直线距离
{
    return sqrt(pow(load[i].x-load[j].x,2)+pow(load[i].y-load[j].y,2));
}
void prim()
{
    double sum=0;//最小生成树的权值和
    for(int i=0; i<n; i++)//初始化所有的点都没有使用过。
        used[i]=false;
    for(int i=0; i<n; i++)//首次赋值。
        mincost[i]=road[0][i];
    ok=used[0]=true;
    for(int ii=0; ii<n-1; ii++)
    {
        double mins=inf;
        int pos=0;
        for(int i=0; i<n; i++)
            if(!used[i]&&mins>mincost[i])
                mins=mincost[pos=i];//找到最小权值。
        //由于题目有条件限制,两个小岛之间的距离必须在[10,1000]之间,
        //所以如果不满足,则会设置两个小岛之间的距离为一个无限大值(即不可达值)
        //然后在找最小值的时候,则会找不到最小值,所以就没办法建设了,即输出"oh!"。
        if(mins==inf)
        {
            cout<<"oh!"<<endl;
            return ;
        }
        sum+=mins*100;
        //cout<<sum<<endl;
        used[pos]=true;
        for(int i=0; i<n; i++)
            if(!used[i])
                if(mincost[i]>road[pos][i])//更新权值。
                    mincost[i]=road[pos][i];
    }
    cout<<fixed<<setprecision(1)<<sum<<endl;
    return ;
}
int main()
{
    cin>>T;//输入样例个数
    while(T--)
    {
        cin>>n;//输入小岛个数
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++)
                if(i==j)
                    road[i][i]=0;//i到i岛的距离为0
                else
                    road[i][j]=inf;//初始化为“无限大”

        for(int i=0; i<n; i++)
            cin>>load[i].x>>load[i].y;//输入小岛的坐标
        for(int i=0; i<n; i++)
            for(int j=i+1; j<n; j++)
            {
                if(dis(i,j)>=10&&dis(i,j)<=1000&&dis(i,j)<road[i][j])
                    road[i][j]=road[j][i]=dis(i,j);
                //如果i到j岛的距离符合条件要求,那么就进行赋值。
            }
        prim();
    }
    return 0;
}

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。