HDU 1233 （最小生成树）用并查集实现kruskal

最新推荐文章于 2019-03-19 23:07:12 发布

李总LKC

最新推荐文章于 2019-03-19 23:07:12 发布

阅读量566

点赞数

分类专栏：最小生成树文章标签：并查集最小生成树

本文链接：https://blog.csdn.net/u014309268/article/details/22054171

版权

最小生成树专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细介绍了使用最小生成树算法解决特定问题的过程。通过具体的代码实现，展示了如何利用并查集进行边的权值比较与合并，从而找到图中所有顶点构成的最小代价生成树。文中特别注意了细节处理，如数组的正确初始化等。

摘要由CSDN通过智能技术生成

题目很简单，不过有些细节要注意

#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 10000

struct edge1
{
	int x,y;
	int weigh;
}edge[MAX];

int father[MAX];


bool cmp(edge1 a, edge1 b)  
{  
	return a.weigh < b.weigh;  
}  



void init(int n)
{
	for(int i=0;i<=n;i++)       //这里要注意，如果写成i<n，那么就会WA，原因是第N个数没有开辟出节点，看清题意，他是从1...N,我错了半天才找出来
	{
		father[i]=i;
	}
}

int find_set(int x)
{
	if(x==father[x]) return x;
	father[x]=find_set(father[x]);
	return father[x];
}



void union_set(int x,int y)
{
	father[y]=x;
}



void main()
{
	int N,i,a,b, sum,M;
	while(cin>>N)
	{
		if(N==0) break;
		init(N);
		 sum=0;
		 M=N*(N-1)/2;
		for(i=0;i<M;i++)
		{
			cin>>edge[i].x>>edge[i].y>>edge[i].weigh;

		}

		sort(edge,edge+M,cmp);


		for(i=0;i<M;i++)
		{
			a=find_set(edge[i].x);
			b=find_set(edge[i].y);
			if(a!=b)
			{
				union_set(a,b);
				sum+=edge[i].weigh;
			}
		}
		cout<<sum<<endl;
	}
}

这样写可能有些别扭，我们干脆把数组从下标“1”开始存储

#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 10000

struct edge1
{
	int x,y;
	int weigh;
}edge[MAX];

int father[MAX];


bool cmp(edge1 a, edge1 b)  
{  
	return a.weigh < b.weigh;  
}  



void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		father[i]=i;
	}
}

int find_set(int x)
{
	if(x==father[x]) return x;
	father[x]=find_set(father[x]);
	return father[x];
}



void union_set(int x,int y)
{
	father[y]=x;
}



void main()
{
	int N,i,a,b, sum,M;
	while(cin>>N)
	{
		if(N==0) break;
		init(N);
		 sum=0;
		 M=N*(N-1)/2;
		for(i=1;i<=M;i++)
		{
			cin>>edge[i].x>>edge[i].y>>edge[i].weigh;

		}

		sort(edge+1,edge+1+M,cmp);


		for(i=1;i<=M;i++)
		{
			a=find_set(edge[i].x);
			b=find_set(edge[i].y);
			if(a!=b)
			{
				union_set(a,b);
				sum+=edge[i].weigh;
			}
		}
		cout<<sum<<endl;
	}
}

下面附上用冒泡法写的算法，不过没有AC过，超时间

#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 10000

struct edge1
{
	int x,y;
	int weigh;
}edge[MAX];

int father[MAX];


bool cmp(edge1 a, edge1 b)  
{  
	return a.weigh < b.weigh;  
}  



void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		father[i]=i;
	}
}

int find_set(int x)
{
	if(x==father[x]) return x;
	father[x]=find_set(father[x]);
	return father[x];
}



void union_set(int x,int y)
{
	father[y]=x;
}



void main()
{
	int N,i,a,b, sum,M;
	while(cin>>N)
	{
		if(N==0) break;
		init(N);
		 sum=0;
		 M=N*(N-1)/2;
		for(i=1;i<=M;i++)
		{
			cin>>edge[i].x>>edge[i].y>>edge[i].weigh;

		}

		sort(edge+1,edge+1+M,cmp);


		for(i=1;i<=M;i++)
		{
			a=find_set(edge[i].x);
			b=find_set(edge[i].y);
			if(a!=b)
			{
				union_set(a,b);
				sum+=edge[i].weigh;
			}
		}
		cout<<sum<<endl;
	}
}