Kruskal算法小笔记

转一个大佬的并查集详解文章 ： http://blog.csdn.net/the_best_man/article/details/62416938

下面是自己的理解  ： 

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
using namespace std;

const int N = 0x3f3f3f3f;
const int M = 0x3f3f3f3f;//根据实际情况变化。
typedef long long ll;
int n, cnt, fa[N];

struct Node{
    int a, b, len;//记录两点及其之间的边权。
}p[M];

void init() {//初始化每个点的上级。
    cnt = 0;
    for (int i = 0; i <= n; i++) {
        fa[i] = i;
    
    }
}

int cmp(Node x, Node y) {
    return x.len < y.len;       //简单的排序函数。
}

int find(int x) {
    return x == fa[x] ? x : fa[x] = find(fa[x]);//查找x的上级。这个三目运算符用的很巧妙。
}

int kruskal() {
    int rec = 0;
    sort(p, p + cnt, cmp);//按边大小进行排序
    double ans = 0;
    for (int i = 0; i < cnt; i++) {
        if (rec == n - 1) break;//因为有n个点，所以只用找n-1次就可以了。
        int x = find(p[i].a), y = find(p[i].b);
        if (x != y) {//如果两点上级不同，即两点不构成回路，ans+。
            fa[x] = y;
            ans += p[i].len;
            rec++;//记录当前找寻次数。
        }
    }
    return ans;
}

void input() {//输入函数。
    int a, b, s;
    for (int i = 0; i < n - 1; i++){
            scanf("%d %d %d", &a, &b, &s);
            p[cnt].a = a;
            p[cnt].b = b;
            p[cnt++].len = b;
    }
}

int main() {
    while (scanf("%d\n", &n) != EOF) {
        if (!n) break;//n为0时结束程序。
        init();
        input();
        printf("%d\n", kruskal());
    }
    return 0;
}