kruskal算法
最近上离散数学学到了krusklal(克鲁斯卡尔)算法,于是找到了一个以前写过的板子,来发出来康康
核心在于书上的几步
- 先将所有的线按照权值排序(快排) o(mlogm)
- 从最小权值开始走,当线段为点-1时,结束我的循环
- 枚举每条边ab,权重为c,如果当前ab不连通,将ab加到集合中(并查集的简单应用),过程中我需要找到这点的父节点,然后用父节点去搞搞
- 每次i++
需要了解的地方,由于咱用的是c,板子其实是c++,在排序上c++要简单一点,详情去了解stl,这个是重载
核心这么写:
const int N = 1000010;
struct Edge{
int a, b, w;
bool operator <(const Edge &W)const {
return w < W.w;
}
}egdes[N];
以上代码用于排序
然后并查集用于查找根节点也就是连接,整体板子如下,我就不多说了
#include<bits/stdc++.h>
using namespace std;
const int N=200010;
int n,m;
int p[N];
struct Edge{
int a,b,w;
bool operator< (const Edge &W)const{ //重载运算符,从小到大排序
return w < W.w;
}
}edges[N];
int find(int x){
if(p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main() {
scanf("%d%d",&n,&m);
for(int i = 0 ; i < m ; i++) {
int a,b,w;
scanf("%d%d%d",&a,&b,&w);
edges[i] = {a,b,w};
}
sort(edges,edges + m); //进行权值排序
for(int i = 1; i <= n; i++) p[i] = i; //点操作,构建图
for(int i = 0; i < m ; i++) {
int a, b, w;
a = edges[i].a, b = edges[i].b, w = edges[i].w;
//找到ab的根节点
a = find(a), b = find(b);
//如果根节点不相同,则连起来,走一次无回路
int res = 0, cnt = 0; //res权值,cnt边
if(a != b) {
p[a] = b;//连接
cnt ++;
res += w;
}
}
if(cnt == n - 1) printf("%d",res);
//如果此时走完了,那么我输出我的最小权值
}
Thats 最小生成树
阿弥陀佛!感冒中…