学习总结——2.15

题目：P3366 【模板】最小生成树 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct Edge {
    int u, v, w;
} Edge;

int cmp(const void *a, const void *b) {
    return ((Edge *)a)->w - ((Edge *)b)->w;
}

int find(int x, int fa[]) {
    while (x != fa[x]) {
        x = fa[x] = fa[fa[x]];
    }
    return x;
}

void kruskal(int n, int m, Edge edge[]) {
    qsort(edge, m, sizeof(Edge), cmp);
    int ans = 0, eu, ev, cnt = 0;
    int fa[n + 1];
    for (int i = 1; i <= n; i++) {
        fa[i] = i;
    }
    for (int i = 0; i < m; i++) {
        eu = find(edge[i].u, fa);
        ev = find(edge[i].v, fa);
        if (eu == ev) {
            continue;
        }
        ans += edge[i].w;
        fa[ev] = eu;
        if (++cnt == n - 1) {
            break;
        }
    }
    printf("%d", ans);
}

int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    Edge edge[m];
    for (int i = 0; i < m; i++) {
        scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w);
    }
    kruskal(n, m, edge);
    return 0;
}

• 首先，使用qsort函数对所有边按权重从小到大排序。
• 初始化并查集，每个节点自成一个集合。
• 遍历排序后的边，对于每条边，检查它的两个顶点是否属于同一个集合。如果不属于同一个集合，则将它们合并，并将这条边的权重加到答案中。当已经选取了n-1条边时（n是顶点的数量），算法结束。
• 输出最小生成树的权重和