ACM-最小生成树之还是畅通工程——hdu1233

还是畅通工程

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

 

Sample Output

3
5


一道典型的最小生成树的题目，此处我用的是适合边数多得Kruskal算法生成最小生成树，这个算法
复杂度主要与排序算法相关，这里我用的选择，比较慢。。  用快排速度大约为 O(N log N)

#include <iostream>
using namespace std;

//树结构体
struct tree
{
	int s1,s2;
	int len;
}tr[10001];

//父数组
int parent[10001];


//排序函数（选择）
void px(int n)
{
	int i,j;
	tree temp;
	for(i=0;i<n-1;++i)
		for(j=i+1;j<n;++j)
			if(tr[i].len>tr[j].len)
			{
				temp=tr[i];
				tr[i]=tr[j];
				tr[j]=temp;
			}
}

// 父函数，返回最顶父节点
int pa(int m)
{
	while(parent[m]!=m)
	{
		m=parent[m];
	}
	return m;
}

// 主函数
void main()
{
	int point,sum=0;
	int i,j;
	int x,y;
	tree tre[10001];

	while(cin>>point && point)
	{
		for(i=0;i<point*(point-1)/2;++i)
			cin>>tr[i].s1>>tr[i].s2>>tr[i].len;
		
		for(i=1;i<=point;++i)
			parent[i]=i;
		
		px(point*(point-1)/2);


		j=0;
		for(i=0;i<point*(point-1)/2;++i)
		{
			x=pa(tr[i].s1);
			y=pa(tr[i].s2);
			
			
			if(x==y)
				continue;
			
			parent[y]=x;
			tre[j]=tr[i];
			++j;
		}
		
		sum=0;
		for(i=0;i<j;++i)
			sum+=tre[i].len;
		cout<<sum<<endl;
	}
}