题意:给n个岛屿的坐标,当两座岛之间的距离超过1000或小于10时不能建桥,而建桥的费用是100/m,求出联通这n座岛的最小费用,如果无法联通,输出“oh”。
这道题只要将距离求出来,将距离超过1000或小于10的设为无穷大进行存储。再将prim算法设成double型,就可以求出来了。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define inf 0x7fffffff
#define arr 105
double g[arr][arr],dis[arr];
struct node{
int x;
int y;
}a[arr];
double caldis(int i,int j)
{
double dist=hypot(a[i].x-a[j].x,a[i].y-a[j].y);//hypot是求直角三角形斜边的函数
if(dist>1000||dist<10)
dist=inf;
return dist;
}
double prim(int st,int m)
{
int i,Time,mark[arr];
for(i=1;i<=m;i++) dis[i]=g[st][i],mark[i]=0;
dis[st]=inf, mark[st]=1;
double ans=0;
for(Time=1;Time<m;Time++){
double min= inf;
int index=-1;
for(i=1;i<=m;i++)
if(!mark[i]&&dis[i]<min) min=dis[i],index=i;
if(index==-1) return -1; //无法生成一个树;
mark[index]=1;
ans+= min;
for(i=1;i<=m;i++)
if(!mark[i]&&g[index][i]<dis[i])
dis[i]=g[index][i];
}
return ans*100.0;
}
int main()
{
int t;
scanf("%d",&t);
int i,j;
while(t--)
{
int n;
scanf("%d",&n);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
g[i][j]=inf;
for(i=1;i<=n;i++)
scanf("%d%d",&a[i].x,&a[i].y);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
{
g[i][j]=g[j][i]=caldis(i,j);
}
double b=prim(1,n);
if(b==-1) printf("oh!\n");
else printf("%.1lf\n",b);
}
return 0;
}