畅通工程再续 最小生成树

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组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!

两个办法，一个是暴力用DFS进行搜索，但是写出来会超时，另一个办法是把点转化为图，用求最小生成树的办法，给出的点构成了一张完全图，求最小距离相当于求最小生成树。另外在求最小生成树的办法时要注意数组的大小，给的点不超过100个，也就是说在存储边的数组中元素个数最多会有100X100即10000个，不要两个都开成100的大小，不然会WA，其中找最小生成树的时候需要用到并查集的知识，如果找的这条边涉及的两个点不在同一个集合里面，这种情况下可以把点加进去，如果两个点已经在一个集合里面了，此时说明再加就会出现回路，不进行合并操作。

DFS(TLE)

#include<stdio.h>
#include<iostream>
#include<memory.h>
#include<math.h>
using namespace std;
struct Node{
	int x;
	int y;
};
struct Node node[1005];
int vis[1005];	
int N,T;
double ans=999999;
double fun(int tx,int ty)
{
	double temp=sqrt((node[tx].x-node[ty].x)*(node[tx].x-node[ty].x)+(node[tx].y-node[ty].y)*(node[tx].y-node[ty].y));
	return temp;
}
void dfs(int x,int cnt,double dis)
{
	if(cnt==N)
	{
		if(dis<ans)
		ans=dis;
		return;
	}
	for(int i=0;i<N;i++)
	{
		if(vis[i]==1) continue;
		double temp=fun(x,i);
		if(temp<10||temp>=1000) continue;
		vis[i]=1;
		dfs(i,cnt+1,dis+temp);
		vis[i]=0;
	}
}
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&N);
		ans=999999;
		memset(vis,0,sizeof(vis));
		for(int i=0;i<N;i++)
			scanf("%d %d",&node[i].x,&node[i].y);
		for(int i=0;i<N;i++)
		{
			vis[i]=1;
			dfs(i,1,0);
			vis[i]=0;
		}
		if(ans!=999999)
		printf("%.1lf\n",100*ans);
		else
		printf("oh!\n");
	}
	return 0;
}

AC代码:

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
int N,T;
struct Node{
	int x;
	int y;
	double dis;
};
int pre[105];
struct Node node[105];
struct Node nodes[10005];
int find(int x)
{
	if(x==pre[x])
	return x;
	else
	return pre[x]=find(pre[x]);
 } 
double fun(int tx,int ty)
{
	return temp=sqrt((node[tx].x-node[ty].x)*(node[tx].x-node[ty].x)+(node[tx].y-node[ty].y)*(node[tx].y-node[ty].y));
	
}
bool cmp(struct Node x,struct Node y)
{
	return x.dis<y.dis;
}
int main()
{
	cin>>T;
	while(T--)
	{
		cin>>N;
		for(int i=0;i<N;i++)
		{
			cin>>node[i].x>>node[i].y;
			pre[i]=i;
		}
		int k=0;
		for(int i=0;i<N;i++)
		{
			for(int j=0;j<i;j++)
			{
				nodes[k].x=i;
				nodes[k].y=j;
				nodes[k].dis=fun(i,j);
				k++;
			}
		}	
		sort(nodes,nodes+k,cmp);
		int ans=0;
		double sum=0;
		for(int i=0;i<k;i++)
		{
			if(nodes[i].dis>=10&&nodes[i].dis<1000)
			{
				int fx=find(nodes[i].x);
				int fy=find(nodes[i].y);
				if(fx!=fy)
				{
					pre[fx]=fy;
					ans++;
					sum+=nodes[i].dis;
				}
			}
			if(ans==N-1) break;
		}
		if(ans==N-1)
		printf("%.1lf\n",sum*100);
		else
		printf("oh!\n");
	}
	return 0; 
}