C++数据结构与算法分析——Kruskal算法

Kruskal算法

介绍

Kruskal算法也是基于贪心算法证得的。虽然Kruskal算法与Prim算法都是求最小生成树的算法，但是它们的操作不太一样。Prim算法是先找到一个点作为一个集合，找到离它最近的点，将它加入集合。而Kruskal则是将所有边的边权从小到大排序，然后遍历所有边，如果两个点不在同一集合内，则将两个点加入到同一集合，如果两个点已经遍历过，则不操作。最后如果集合内的边数为n - 1则说明生成了最小生成树，否则没有。

说明

如何将两个点加入到同一集合呢？在并查集中已经说过了，并查集可以完成集合的合并和集合的查询，因此我们可以用并查集来实现将两个边加入到同一集合的过程。

步骤

1. 建立并查集，每个点各自构成一个集合。
2. 将所有边按边权从小到大排序，依次扫描每条边(a,b,c)。
3. 若a,b已属于同一集合，则跳过，扫描下一条。
4. 否则，合并a,b两个集合，并把边权c加入答案，边数+1
5. 所有边扫描完成后，判断边数是否等于n - 1，若相等则得出了最小生成树。

例题

题目描述

给定一个n个点m条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

给定一张边带权的无向图 G=(V,E)，其中 V 表示图中点的集合，E 表示图中边的集合，n=|V|，m=|E|。

由 V 中的全部 n 个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G 的一棵生成树，其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。

输入格式

第一行包含两个整数n和m。

接下来m行，每行包含三个整数u,v,w，表示点u和点v之间存在一条权值为w的边。

输出格式

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

数据范围

$1\le n\le 10^5$,
$1\le m\le 2\ast 10^5$,
图中涉及边的边权的绝对值均不超过 1000。

输入样例：

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4

输出样例：

6

解题思路

本题并不需要存边的矩阵或邻接表，只需要一个存边结构体和并查集即可。

代码

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 2e5 + 10;
int n,m;
int p[N]; // 并查集的父节点

struct Edge{ // 定义边
    int a,b,w; // 连通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}; // 将边存入结构体
    }
    
    for(int i = 1; i <= n; i ++) p[i] = i; // 初始化并查集
    
    sort(edges,edges + m); // 对m条边进行从小到大排序
    
    int res = 0,cnt = 0; // res为最小生成树的总权值，cnt为已加入最小生成树的点的个数
    for(int i = 0; i < m; i ++){
        int a = edges[i].a,b = edges[i].b,w = edges[i].w;
        a = find(a),b = find(b);
        if(a != b){
            p[a] = b; // 将a,b连通
            res += w; // 最小生成树总权值 + w
            cnt ++; // 边数 + 1
        }
    }
    
    if(cnt < n - 1) puts("impossible"); // 假如最后未连接n - 1条边，说明图不连通
    else printf("%d",res); // 若图连通则输出答案
    
    return 0;
}