https://cn.vjudge.net/problem/HDU-1875
相信大家都听说一个“百岛湖”的地方吧,百岛湖的居民生活在不同的小岛中,当他们想去其他的小岛时都要通过划小船来实现。现在*决定大力发展百岛湖,发展首先要解决的问题当然是交通问题,*决定实现百岛湖的全畅通!经过考察小组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<iostream>
#include<stdio.h>
#include<bits/stdc++.h>
const int inf=;
using namespace std;
int x[],y[];
double w[][],dis[];
int n,m;
bool vis[];
double weight(int i,int j)
{
return sqrt(1.0*((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])));
}
void prim()
{
int i,j,k;
bool flag=;
double sum=,tmp;
for(int i=;i<=m;i++)
dis[i]=inf;
dis[]=;
for(i=;i<=m;i++)
{
tmp=inf;
for(j=;j<=m;j++)
if(!vis[j]&&tmp>dis[j])
{
tmp=dis[j];
k=j;
}
if(tmp==inf){flag=;break;}
vis[k]=;
sum+=tmp;
int y;
for(j=;j<=m;j++)
{
if(!vis[j]&&dis[j]>w[k][j])
dis[j]=w[k][j];
} }
if(flag)
printf("%.1f\n",sum*);
else
printf("oh!\n");
} int main()
{
cin>>n;
for(int i=;i<n;i++)
{
cin>>m;
memset(x,,sizeof x);
memset(y,,sizeof y);
memset(vis,,sizeof vis);
for(int j=;j<=m;j++)
cin>>x[j]>>y[j];
for(int k=;k<=m;k++)
for(int l=;l<=m;l++)
{
double p=weight(k,l);
if(p>=&&p<=)
w[k][l]=w[l][k]=p;
else
w[k][l]=w[l][k]=inf;
}
prim();
}
return ;
}
思路:
使用Prim求解最小生成树。
注意点:
1.一定要记得给每个数组初始化,一开始忘记给vis[]初始化了,导致调试了好久。
2.用memset的时候要注意类型,我给double赋极大值错误地使用了memset(dis,0x3f,sizeof dis);后来想到double类型是8个字节的,int是四个字节的(这是四字节赋极大值的方法,赋给8个字节的double类型的数得到的结果不一样),因为memset是按字节赋值的。这么给double赋值得到了一个约等于0的数,与预想的结果不同,导致错误。
Prim模板(解决最小生成树问题):
void prim()
{
int i,j,k,tmp;
memset(dis,0x3f,sizeof dis);
dis[]=;
for(i=;i<=n;i++)
{
tmp=inf;
for(j=;j<=n;j++)
if(!vis[j]&&tmp>dis[j])
{
tmp=dis[j];
k=j;
}//找到最小距离的节点
vis[k]=;
for(j=;j<=n;j++)
{
if(!vis[j]&&dis[j]>g[k][j])
dis[j]=g[k][j];
}//更新最短距离
}
}