HDU ACM 1233 还是畅通工程

标签：算法   acm   编程   c++   c   

畅通工程的升级版。

求最小生成树。Kruskal 算法：应用并查集。

#include<iostream>   
#include<algorithm>
using namespace std;


int p[102];


bool Init(int n)                //一开始指向自己
{
	for(int i=0;i<=n;i++)
		p[i]=i;
	return true;
}


int Find(int x)            //找到根节点
{
	int r,i,j;


	r=x;
	while(r!=p[r]) r=p[r];      //路径压缩准备,找到根节点


	i=x;
	while(i!=p[i])
	{
		j=p[i];
		p[i]=r;
		i=j;
	}
	return i;                 //返回根节点
}


bool Merge(int x,int y)      //返回是否需要合并
{
	int tx,ty;


	tx=Find(x);
	ty=Find(y);


	if(tx==ty) return false;
	p[tx]=ty;
	return true;
}


struct Path
{
	int a,b;
	int v;
};


bool cmp(Path a,Path b)
{
	return a.v<b.v;
}


int main()  
{ 
	int n,i,sum,cnt;
	Path pa[5002];


	while(scanf("%d",&n)==1&& n)
	{
		for(i=0;i<n*(n-1)/2;i++)
			scanf("%d%d%d",&pa[i].a,&pa[i].b,&pa[i].v);


		sort(pa,pa+n*(n-1)/2,cmp);             //根据路劲从小到大排序
		sum=0;
		cnt=1;
		Init(n);
		for(i=0;cnt<n;i++)                   //找到n-1一条路
			if(Merge(pa[i].a,pa[i].b))
			{
				sum+=pa[i].v;
				cnt++;
			}


		printf("%d\n",sum);
	}
    return 0;  
}

HDU ACM 1233 还是畅通工程

标签：算法   acm   编程   c++   c   

原文地址：http://blog.csdn.net/a809146548/article/details/44276761