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