2009-1.17-pku2560.txt

MST算法就是最小生成树算法！ 在ACM中这个应该是比较简单的一个算法！ 大家好好学习吧！

#include<stdio.h>
#include<algorithm>
#include<math.h>
using namespace std;
const int MAXE=6005;
const int MAX=105;
int set[MAX];
int rank[MAX];
typedef struct Edge
{
	int st;//边的起点
	int ed;//边的终点
	double distance;//边的权值
}Edge;
Edge edge[MAXE];
typedef struct Point
{
	double x;
	double y;
}Point;
Point p[MAX];
/*下面三个函数是求并查集的函数!*/
//第一函数是初始化
void Make_Set(int x)
{
	set[x]=x;
	rank[x]=0;
}
//返回包含x的集合中的一个代表元素
int Find_Set(int x)
{
	if(x!=set[x])
		set[x]=Find_Set(set[x]);
	return set[x];
}

//实现树与树的合并
void Union(int x,int y)
{
	x=Find_Set(x);
	y=Find_Set(y);
	if(rank[x]>rank[y])
		set[y]=x;
	else
	{
		set[x]=y;
		if(rank[x]==rank[y])
			rank[y]++;
	}
}
bool cmp(Edge a,Edge b)
{
	return a.distance<b.distance;
}
double Distance(Point a,Point b)
{
	double dis;
	dis=(a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);//NN的这个函数开始写错了，搞了我WA了N次了!
	return sqrt(dis);
}
/*关键函数-Kruskal算法的实现!*/
double Kruskal(int V,int E)
{
	int i;
	double sum=0.0;
	for(i=1;i<=V;i++)
		Make_Set(i);//每个点构成一个树也即一个集合
	sort(edge+1,edge+E+1,cmp);//将边按照权值非降序排序
	for(i=1;i<=E;i++)
		if(Find_Set(edge[i].st)!=Find_Set(edge[i].ed))
		{//如果是安全边就加sum中去
			sum+=edge[i].distance;
			//并将包含这两棵树的集合合并
			Union(edge[i].st,edge[i].ed);
		}
	return sum;
}
/*求最小生成树的Kruskal算法 按照<<算法导论>>自己写了一个模板!*/
int main()
{
	int V,E,i,j;
	double sum;
	while(scanf("%d",&V)!=EOF)
	{
		E=0;
		for(i=1;i<=V;i++)
			scanf("%lf%lf",&p[i].x,&p[i].y);
		for(i=1;i<=V-1;i++)
			for(j=i+1;j<=V;j++)
			{
				edge[++E].st=i;
				edge[E].ed=j;
				edge[E].distance=Distance(p[i],p[j]);
			}
		sum=0;
		sum=Kruskal(V,E);
		printf("%.2lf\n",sum);
	}
	return 0;
}
/*
3
1.0 1.0
2.0 2.0
2.0 4.0

3.41
*/