Kruskal算法的核心思想就是将边的权值从小到大进行枚举,枚举的过程中看该边的两点是否连通,假如不连通,那么这条边就可以加入到最小生成树中,假如连通,再将其加入最小生成树中,会产生回路。
假如生成最小生成树到了上图这个时刻,此时有还有两条边的信息,4<-->5 权值为7, 4<--->3 权值为9,所以此时可看出4,5是连通的,应该忽略进行下一轮,此时4<-->3权值为9,可知3,4两点不连通,加入到最小生成树中。
此时加入到最小生成树中已经n-1条边了,生成最小生成树结束。
对了,应该怎么判断是否连通呢?并查集!不知道的朋友可以看上一篇博文。
#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
struct node{
int a, b, w;
node(int a, int b, int w){
this->a = a;
this->b = b;
this->w = w;
}
friend bool operator<(node n1, node n2){
return n1.w > n2.w;
}
};
int f[105], book[105] = {0};
void init(int n){
int i;
for(i = 1; i <= n; ++i)
f[i] = i;
}
int getF(int x){
if(x == f[x]) return x;
else{
f[x] = getF(f[x]);
return f[x];
}
}
int merge(int a, int b){
int fa, fb;
fa = getF(a);
fb = getF(b);
if(fa != fb) {
f[fb] = fa;
return 0;
}
return 1;
}
int main(){
int n, m, i, a, b, w, k, ans;
while(cin >> n >> m){
priority_queue<node> q;
for(i = 1; i <= m; ++i){
cin >> a >> b >> w;
q.push(node(a, b, w));
}
init(n);
k = 1, ans = 0;
while(k < n){
node x = q.top();
q.pop();
if(merge(x.a, x.b)) continue;
ans += x.w;
++k;
}
printf("%d\n", ans);
}
return 0;
}
Kruskal的时间复杂度,若通过排序做,一是进行快速排序O(M logM), 寻找n-1条边是O(N logM),总的时间复杂度为O(MlogM + NlogM),通常M大于N,所以时间复杂度可认为是O(MlogM)。
在这儿博主用了优先队列,更加优化了一些,时间复杂度也为O(N logM),但省去了一些排序方式的开销。