HDOJ 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): 16624    Accepted Submission(s): 5182


[align=left]Problem Description[/align]
相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。
 

[align=left]Input[/align]
输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。

每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

 

[align=left]Output[/align]
每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.
 

[align=left]Sample Input[/align]

2
2
10 10
20 20
3
1 1
2 2
1000 1000

 

[align=left]Sample Output[/align]

1414.2
oh!

 

#include<stdio.h>
#include<string.h>
#include<math.h>
#define fun(x1,x2,y1,y2) sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2))
#define INT_MAX 100000000
#define NUM 110
struct Seat
{//保存各岛的坐标
double x,y;
}num[NUM];
double cost[NUM][NUM],sum;//cost[i][j]表示i岛与j岛之间的距离。如果可以连通，sum为总长度
bool judge[NUM];//判断各个岛是否在最小生成树的集合中
double prim(int n)
{
int i,j,pos;//pos表示当前集合外到集合最小距离的岛
double Long[NUM],sum=0,min;//Long[]表示当前各集合外的点到集合的最小距离
memset(judge,false,sizeof(judge));
pos=1;//开始选择岛1为起始岛（点）
judge[pos]=true;//将岛1放到集合中
for(i=1;i<=n;i++)
if(i!=pos) Long[i]=cost[pos][i];//此时岛1为集合内的唯一岛，岛1到集合外各岛的距离
for(i=0;i<n-1;i++)
{
min=INT_MAX;
for(j=1;j<=n;j++)//找出当前集合外一点pos，该点到集合的最短距离min
if(!judge[j]&&Long[j]<min)
{
min=Long[j];
pos=j;
}
if(min==INT_MAX) return -1;//如果当前集合到集合外的距离都是无穷远，则不能连通；
sum+=min;
judge[pos]=true;//将pos加入集合中
for(j=1;j<=n;j++)//加入pos点后，更新集合到集合外各点的距离
if(!judge[j]&&cost[pos][j]<Long[j])
Long[j]=cost[pos][j];
}
return sum;
}
int main()
{
int T,N,M,i,j;
scanf("%d",&T);//T组数据
while(T--)
{
scanf("%d",&N);//N个小岛
for(i=1;i<=N;i++)//1~N小岛的坐标
scanf("%lf %lf",&num[i].x,&num[i].y);
for(i=1;i<=N;i++)//初始化
for(j=1;j<=N;j++)
if(i==j) cost[i][j]=0;//如果i和j相等，赋0
else cost[i][j]=INT_MAX;//否则，赋值为无穷
for(i=1;i<=N;i++)//计算i和j两岛之间的距离
for(j=1;j<=N;j++)
{
cost[i][j]=fun(num[i].x,num[j].x,num[i].y,num[j].y);//i点与j点间的距离
if(!(cost[i][j]>=10&&cost[i][j]<=1000))//不满足距离大于等于10并且小于等于1000的情况
{
if(i==j) cost[i][j]=0;
else cost[i][j]=INT_MAX;
}
}

if(prim(N)==-1)//当返回值为-1时，表示不连通
printf("oh!\n");
else
printf("%.1lf\n",prim(N)*100);

}
return 0;
}