克鲁斯卡尔算法思想:
1、将所有边权从小到大排序
2、按照边权从小到大测试所有的边,如果该边的两个顶点不在同一个连通块中,则把这条测试边加入当前最小生成树中。
3、执行步骤2,知道最小生成树的边数等于顶点数-1,如果结束时最小生成树的边数小于顶点数-1,说明该图不连通。
注意步骤2:
测试两个顶点在不在一个连通块中,如果把连通块看成一个集合,那么就转化成判断两个顶点是否在同一个集合中,所以可以使用并查集。将测试边加入最小生成树,则可以用并查集的合并操作。即只要把测试边的两个端点所在的集合合并,就能达到将边加入最小生成树的效果
假设顶点的编号范围是[0,n-1](并查集初始化时范围不能弄错)。如果下标变化,只需要修改并查集初始化部分即可。
#include<bits/stdc++.h>
using namespace std;
const int MAXV = 110;
const int MAXE = 10010;
struct edge
{
int u, v; //边的两个端点的编号
int cost; //边权
}E[MAXE]; //最多有MAXE条边
bool cmp(edge a, edge b)
{
return a.cost < b.cost;
}
// 并查集部分
int Father[MAXV];
int findFather(int x)
{
//寻找根结点
int a = x;
while(x != Father[x])
x = Father[x];
//路径压缩
while(a != Father[a])
{
int z = a;
a = Father[a];
Father[z] = x;
}
return x;
}
int kruskal(int n, int m) //n为顶点数,m为边数
{
int ans = 0, Num_Edge = 0; //ans为权值和 Num_Edge为当前生成树的边数
for(int i = 0; i < n; i++) // 顶点范围是[0,n-1]
Father[i] = i; //并查集初始化
sort(E, E + m, cmp); //所有边按权重从小到大排序
for(int i = 0; i < m; i++) //枚举所有边
{
int faU = findFather(E[i].u); //查询两个端点所在集合的根结点
int faV = findFather(E[i].v);
if(faU != faV) //如果不在一个集合中
{
Father[faU] = faV; //合并集合(即把测试边加入最小生成树)
// Father[faV] = faU; 写这句也可以
ans += E[i].cost; //边权之和增加测试边的边权
Num_Edge++; // 当前生成树的边数+1
if(Num_Edge == n - 1) //边数等于顶点数-1时结束算法
break;
}
}
if(Num_Edge != n - 1) //无法连通时返回-1
return -1;
else
return ans; //返回最小生成树的边权之和
}
int main()
{
int n, m; //顶点数,边数
scanf("%d%d", &n, &m);
for(int i = 0; i < m; i++)
scanf("%d%d%d", &E[i].u, &E[i].v, &E[i].cost); //两个顶点编号,边权
int ans = kruskal(n, m);
printf("%d\n", ans);
}
输入
6 10
0 1 4
0 4 1
0 5 2
1 2 6
1 5 3
2 3 6
2 5 5
3 4 4
3 5 5
4 5 3
输出
15