HDU-1875-畅通工程再续

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1875

畅通工程再续

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!
  
  

 

prim算法 代码实现如下：

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int n;
double e[111][111],dis[111];
bool visit[111];
int flag=0;
const double inf=9999999;//double型
void init()
{
	for(int i=1;i<=n;i++)  
    {  
        for(int j=1;j<=n;j++)  
        {  
            if(i==j)  
            e[i][j]=0;  
            else  
            e[i][j]=inf;  
        }  
    } 
}
void prim()
{
	int i,j,k;
	double minn;
    double sum=0;
	for(i=1;i<=n;i++)
	{
		dis[i]=e[1][i];
	} 
	memset(visit,0,sizeof(visit));
	visit[1]=1;//标记已加入。 
	int count=1;
	while(count<n)
	{
		minn=inf;
		for(i=1;i<=n;i++)
		{
			if(visit[i]==0&&dis[i]<minn)
			{
				minn=dis[i];
				j=i;
			}
		}
		//不能连通。 
		if(minn==inf)
		{
			printf("oh!\n");
			return;
		}
		visit[j]=1;
		count++;
		sum=sum+dis[j];
		for(k=1;k<=n;k++)
		{
			if(visit[k]==0&&dis[k]>e[j][k])
			dis[k]=e[j][k];
		} 
	 } 
	printf("%.1lf\n",100*sum);
} 
int main()
{
	int T,i,j;
	cin>>T;
	while(T--)
	{
		int x[111],y[111];
		cin>>n;
		init();
		for(i=1;i<=n;i++)
		scanf("%d%d",&x[i],&y[i]);
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				if(i==j)
				continue;
				e[i][j]=sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
				if(e[i][j]>1000||e[i][j]<10)
				e[i][j]=inf;
			}
		}
		prim();		
	}
	return 0;
 }

kruskal 算法代码如下：

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n;
int pre[111];
struct edge{
	int u,v;
	double w;
}e[111111];//把数组开大 
void init()  
{  
    for(int i=1;i<=n;i++)  
    {  
        pre[i]=i;  
    }  
}    
int find(int v)  
{  
    if(pre[v]==v)  
        return v;  
    else  
    {  
        pre[v]=find(pre[v]);  
        return pre[v];  
    }  
}   
int merge(int u,int v)  
{  
    int t1=find(u);  
    int t2=find(v);  
    if(t1!=t2)   
    {  
        pre[t2]=t1;  
        return 1;   
    }  
    return 0;  
}     
bool cmp(edge a,edge b)  
{  
    return a.w<b.w;  
}   
int main()
{
	int T;
	int i,j,x[111],y[111];
	double sum;
	scanf("%d",&T);
	while(T--)
	{
		cin>>n;
		for(i=1;i<=n;i++)
		{
			scanf("%d%d",&x[i],&y[i]);
		}
		int m=1;
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=i;j++)//这里存半张图就够了
			{
				double ww=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
				if(ww>=10&&ww<=1000)
				{
					e[m].u=i;
					e[m].v=j;
					e[m].w=ww;
					m++;
				}
			}
		}
		if(m==1)
		{
			printf("oh!\n");
			continue;
		}
		sort(e+1,e+m+1,cmp);
		init();
		int count=0;
		sum=0.0;  
	    for(i=1;i<=m;i++)
	    {  
	        if(merge(e[i].u,e[i].v))  
	        {  
	            count++;  
	            sum=sum+e[i].w;  
	        }  
	    }
	    if(count==n-1)
	    printf("%.1lf\n",sum*100);
	    else
	    printf("oh!\n");    
	}
	return 0;
}