hdu 畅通工程再续（最小生成树）（Prim算法 && Kruskal算法）

畅通工程再续

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!

代码（Kruskal算法）：

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;

#define maxn 100+10
#define maxv 10000+10
struct edge
{
int u,v;
double w;
} e[maxv];
int pre[maxn],x[maxn],y[maxn];
int n,len;

bool cmp(edge a,edge b)
{
return a.w<b.w;
}

void init()
{
for(int i=1; i<maxn; i++)
pre[i]=i;
}

int Find(int x)
{
if(x==pre[x])
return x;
else
{
pre[x]=Find(pre[x]);
return pre[x];
}
}

int join(int x,int y)
{
int fx=Find(x),fy=Find(y);
if(fx!=fy)
{
pre[fy]=fx;
return 1;
}
return 0;
}

void Kruskal()
{
sort(e,e+len,cmp);
int cont=0;
double sum=0;
for(int i=1; i<len; i++)
{
if(e[i].w<10||e[i].w>1000)
continue;
if(join(e[i].u,e[i].v))
{
cont++;
sum+=e[i].w*100;
}
if(cont==n-1)
break;
}
if(cont<n-1)
printf("oh!\n");
else
printf("%.1lf\n",sum);
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
init();
int i,j;
double w;
len=1;
for(i=1; i<=n; i++)
{
scanf("%d%d",&x[i],&y[i]);
for(j=1; j<i; j++)
{
w=sqrt((double)(x[i]-x[j])*(double)(x[i]-x[j])+(double)(y[i]-y[j])*(double)(y[i]-y[j]));
e[len].u=i,e[len].v=j,e[len++].w=w;
}
}
Kruskal();
}
return 0;
}

代码（Prim算法）：

#include<stdio.h>
#include<math.h>
#include<string.h>

#define maxn 100+10
const double inf=0x3f3f3f3f;
double e[maxn][maxn],d[maxn];
bool vis[maxn];
int x[maxn],y[maxn];
int n;

void Prim()
{
int i,j,v;
double minn;
for(i=1; i<=n; i++)
{
d[i]=e[1][i];
vis[i]=0;
}
vis[1]=1;//已经找到了一个点
double sum=0;
for(j=1; j<n; j++)//还需找n-1个点
{
minn=inf;
for(i=1; i<=n; i++)
if(!vis[i]&&d[i]<minn)
minn=d[i],v=i;
if(minn==inf)
{
printf("oh!\n");
return ;
}
vis[v]=1,sum+=d[v];
for(i=1; i<=n; i++)
if(!vis[i]&&d[i]>e[v][i])
d[i]=e[v][i];
}
printf("%.1lf\n",sum*100);
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
int i,j;
double w;
for(i=1; i<=n; i++)
for(j=1; j<=n; j++)
i==j?e[i][j]=0:e[i][j]=e[j][i]=inf;
for(i=1; i<=n; i++)
{
scanf("%d%d",&x[i],&y[i]);
for(j=1; j<i; j++)
{
w=sqrt((double)(x[i]-x[j])*(double)(x[i]-x[j])+(double)(y[i]-y[j])*(double)(y[i]-y[j]));
if(w>1000||w<10)
e[i][j]=e[j][i]=inf;
else
e[i][j]=e[j][i]=w;
}
}
Prim();
}
return 0;
}