/*
* Kruskal算法求MST
* 对边操作,并排序
* 切记:初始化赋值问题(tol)
*/
const int MAXN = 110; // 最大点数
const int MAXM = 10000; // 最大边数
int F[MAXN]; // 并查集使用
struct Edge
{
int u; // 起点
int v; // 终点
int w; // 权值
} edge[MAXM]; // 存储边的信息
int tol; // 边数,加边前赋值为0
void addEdge(int u, int v, int w)
{
edge[tol].u = u;
edge[tol].v = v;
edge[tol++].w = w;
return ;
}
bool cmp(Edge a, Edge b)
{
// 排序函数,将边按照权值从小到大排序
return a.w < b.w;
}
int find(int x)
{
if (F[x] == x)
{
return x;
}
else
{
return F[x] = find(F[x]);
}
}
int Kruskal(int n) // 传入点数,返回最小生成树的权值,如果不连通则返回-1
{
for (int i = 0; i <= n; i++)
{
F[i] = i;
}
sort(edge, edge + tol, cmp);
int cnt = 0; // 计算加入的边数
int ans = 0;
for (int i = 0; i < tol; i++)
{
int u = edge[i].u;
int v = edge[i].v;
int w = edge[i].w;
int tOne = find(u);
int tTwo = find(v);
if (tOne != tTwo)
{
ans += w;
F[tOne] = tTwo;
cnt++;
}
if (cnt == n - 1)
{
break;
}
}
if (cnt < n - 1)
{
return -1; // 不连通
}
else
{
return ans;
}
}
kruskal板子
最新推荐文章于 2022-07-03 12:19:38 发布