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

来源:互联网 发布:淘宝网电脑旺旺打不开 编辑:程序博客网 时间:2024/05/16 18:40

题目地址:[传送门(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代码如下:(代码中有注释。)
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;}
0 0
原创粉丝点击