图文详解最小生成树算法并利用并查集实现Kruskal算法(C++实现)

最小生成树即在连通图中用n-1条边连接全部n个顶点，生成无环路的树形结构，并使得连接边的权值之和最小。
我们介绍最小生成树的经典算法：普里姆算法和克鲁斯卡尔算法。

Prim算法

普里姆算法从图的一个顶点出发，将此顶点归入树的集合，并不断选取图中其他顶点到已添加顶点的最短路径，将添加的最短路径归入最小生成树中，待所有顶点归入集合，即生成n个顶点n-1条边的最小生成树。

//针对图的邻接矩阵的普里姆算法
void MiniSpanTree_Prim(MGraph G)
{
	int min;
	int j,k;
	int adjvex[MAXVEX];
	int lowcost[MAXVEX];
	lowcost[0]=0;
	adjvex[0]=0;
	for (int i = 1; i < G.numVerTexes; i++)
	{
		lowcost[i]=G.arc[0][i];
		adjvex[i]=0;
	}
	for (int i = 1; i < G.numVerTexes; i++)
	{
		min=INFINITY;
		j=1;
		k=0;
		while (j<G.numVerTexes)
		{
			if (lowcost[j]!=0&&lowcost[j]<min)
			{
				min=lowcost[j];
				k=j;
			}
			j++;
		}
		printf("(%d,%d)\n",adjvex[k],k);
		G.arc[adjvex[k]][k];
		lowcost[k]=0;
		for (int j = 1; j < G.numVerTexes; j++)
		{
			if (lowcost[j]!=0&&G.arc[k][j]<lowcost[j])
			{
				lowcost[j]=G.arc[k][j];
				adjvex[j]=k;
			}
		}
	}
}

我们构造如图所示的连通图，然后利用prim算法输出构建最小生成树的过程，我们还引入全局变量来计算最小带权路径长度。

通过输出我们可以看到普里姆算法构建的顺序，最终我们生成了如下图红色路径显示的最小生成树，序号显示构造顺序：

Kruskal算法

克鲁斯卡尔算法理解起来并不难，我们需要一个已经按权值从小到大排好序的边集数组，然后不断从边集数组中选取权值最小的边加入树集合中，只要新添加的边和已在树集合中的边不形成回路，就将其添加到树集中。
最终遍历完边集数组即已经生成最小生成树。

typedef struct
{
	int begin;
	int end;
	int weight;
}Edge;//边集数组
void MiniSpanTree_Kruskal(MGraph G)
{
	int n,m;
	Edge edges[MAXEDGE];//MAXEDGE为最大边数
	InitEdge(edges);//初始化边集数组
	int parent[MAXEDGE];
	for (int i = 0; i < G.numVerTexes; i++)
	{
		parent[i]=0;//初始时parent值都为0
	}
	for (int i = 0; i < G.numEdges; i++)
	{
		n=Find(parent,edges[i].begin);
		m=Find(parent,edges[i].end);
		if (n!=m)
		{
			parent[n]=m;
			printf("(%d,%d) %d\n",edges[i].begin,edges[i].end,edges[i].weight);
		}
	}
}
int Find(int *parent,int f)
{
	while(parent[f]>0)
		f=parent[f];
	return f;
}

还是利用刚才的连通图，然后利用Kruskal算法输出构建最小生成树的过程，我们还引入全局变量来计算最小带权路径长度：
这里边集数组在输入时已经排好序

通过输出我们可以看到克鲁斯卡尔算法构建的顺序，最终我们生成了如下图红色路径显示的最小生成树，序号显示构造顺序：

Kruskal算法与并查集

通过Kruskal算法的构造过程，我们可以看到

n=Find(parent,edges[i].begin);
m=Find(parent,edges[i].end);
if (n!=m)
{
	parent[n]=m;
}
int Find(int *parent,int f)
{
	while(parent[f]>0)
		f=parent[f];
	return f;
}

这里Find函数是返回当前结点的祖先，只要一条边的两个结点祖先不同，则说明这两个节点不相连（不属于同一连通分支），我们就可以把这条边加入最小生成树中而且不会形成环。
这样一查一并的过程就是利用了并查集：图文详解并查集
我们利用并查集维护，重写Kruskal算法的代码：

void MiniSpanTree_Kruskal_UnionFind(MGraph G)
{
	int n,m;
	Edge edges[MAXEDGE];
	InitEdge(edges);
	init_union(G.numVerTexes);//初始化并查集
	for (int i = 0; i < G.numEdges; i++)
	{
		n=find(edges[i].begin);//查
		m=find(edges[i].end);
		if (!same(n,m))//判断是否属于同一连通分支
		{
			unite(n,m);//并
			printf("(%d,%d) %d\n",edges[i].begin,edges[i].end,edges[i].weight);
		}
	}
}

实际上代码可以进一步减少：

void MiniSpanTree_Kruskal_UnionFind(MGraph G)
{
	Edge edges[MAXEDGE];
	InitEdge(edges);
	init_union(G.numVerTexes);
	for (int i = 0; i < G.numEdges; i++)
	{
		if (!same(edges[i].begin,edges[i].end))
		{
			unite(edges[i].begin,edges[i].end);
			printf("(%d,%d) %d\n",edges[i].begin,edges[i].end,edges[i].weight);
		}
	}
}

完整测试代码

完整测试代码：C++代码实现算法及实例