AcWing 859. Kruskal算法求最小生成树

AcWing 859. Kruskal算法求最小生成树

前置问题

Q：什么是生成树呢？
A： 在一个图中，由n个点(其实就是所有顶点)和n-1条边所构成的连通子图(结构是一颗树)

Q：怎么判断哪些点是在同一个连通块？
A：使用并查集就行了，把每个独立的连通块看作是一个集合

思想

Kruskal算法是先把所有边按照边权大小进行排序，得到一个升序的序列，然后遍历每一条边，如果这条边的两个点不是在同一个集合，则选择当前的边权(因为排序过，所以当前边权也是两点之间的最小边权)，顺便合并这两个点所在的集合；如果这两个点已经在同一个集合内，则说明之前已经处理过这两点之间的边权了，不需要再次处理

解决方案

1.输入点数和边数，初始化存储边数的结构体数组，初始化并查集

2.以边权大小为标准将数组进行升序排序

3.对m条边进行处理

3.1 判断这条边的两个点是否属于同一个集合，如果是，则已经处理了

3.2 如果不是，则把两个点所在的集合合并，并记录最小边权，同时也记录合并集合的次数

代码

#include <iostream>
#include <algorithm>

using namespace std;
const int N = 1e5 + 10, M = 2e5 + 10, INF = 0x3f3f3f3f;
//p为并查集数组，n为点数，m为边数，cnt为记录生成树边数的变量，ans为最小生成树的所有边权之和
int p[N], n, m, cnt, ans;

//图一般是使用邻接矩阵或者邻接表或者链式前向星来存储，但因为要对边权排序，所以使用结构体就方便多了
struct Edge {
    //a为边的起点，b为边的终点，c为边权
    int a, b, c;
    
    //运算符重载
    bool operator< (const Edge &e) const {
        return c < e.c;
    }
} edge[M];//声明结构体的同时就创建结构体数组，用来存储图的所有边

//并查集函数，使用了路径压缩
int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

//Kruskal算法的实现
void kruskal() {
	//先对所有边按边权大小进行排序
    sort(edge, edge + m);
    
    //初始化并查集数组，编号为1~n
    for (int i = 1; i <= n; i++) p[i] = i;
    
    //对排序好的边进行处理
    for (int i = 0; i < m; i++) {
        int a = edge[i].a, b = edge[i].b, c = edge[i].c;
        int x = find(a), y = find(b);    //分别找a和b所在的集合的代表节点
        
        /*如果这两点不是连通的，现在这条边就把这两点连通，并记录这条边的权值
          如果这两点是连通的，就说明之前已经处理过 这两点之间的最小权值了，
          因为边的权值是从小到大排序,所以任意两点之间的最小权值的边已经被处理了，
          剩下的这两点之间的其他边就不需要处理了*/
        if (x != y) {     
            p[x] = y;
            ans += c;   //记录最小的权值
            cnt++;
        }
    }

	/*n个集合经过处理如果只剩下一个集合，那么肯定是进行了n-1次操作
	  如果操作次数少于n-1，那么图就不连通 */
    if (cnt < n - 1) puts("impossible");
    else cout << ans;
}

int main() {
    cin >> n >> m;    //输入点数和边数
    for (int i = 0; i < m; i++) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        edge[i] = {a, b, c};
    }
    kruskal();
    return 0;
}