kruskal算法:
kruska算法同样是解决最小生成树的问题的一个算法。和prim算法不同,kruskal算法采用了边贪心的策略。其基本思想为:在初始状态时隐去图中所有的边,这样图中每个顶点都自成一个连通块。之后进行下面的步骤:
- 对所有的边按边权从小到大进行排序。
- 按边权从小到大测试所有的边,如果当前测试边所连接的两个顶点不在同一个连通块中,则把这条测试边加入到当前的最小生成树中;否则,将边舍弃。
- 执行步骤2,直到生成树中的边数等于总顶点数减一或者测试完所有的边时结束。而当结束时,如果最小生成树的边数小于总顶点数减一,说明该图不连通。
下面是一个例子,求出下面这个图的最小生成树
代码:
#include <iostream>
#include<algorithm>;
using namespace std;
const int MAXV = 110;
const int MAXE = 10010;
//边定义部分
struct edge {
int u, v;//边的两个端点的编号
int cost;//边权
}E[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;
}
//kruskal部分,返回最小生成树的边权之和,参数n为顶点的个数,m为图的边数
int kruskal(int n, int m)
{
//ans为所求的边权之和,NUm_Edge为当前生成树的边数
int ans = 0, Num_Edge = 0;
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; //合并集合(即把测试边加入到最小生成树中)
ans += E[i].cost;//边权之和增加测试边的边权
Num_Edge++;//当前生成树的边数加1
if (Num_Edge == n - 1) break;//边数等于顶点数减一时结束算法
}
}
if (Num_Edge != n - 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);//kruskal算法入口
printf("%d\n", ans);
return 0;
}
}