《算法笔记》并查集

并查集

#include <iostream>
#include<vector>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 300;
int father[maxn];//存储父节点
int isRoot[maxn];//存储以这个元素为根节点时集合的元素个数

void init(int n){
    for(int i = 0; i <= n; i++){
        father[i] = i;
        isRoot[i] = 1;
    }
}

int findFather(int a){
    int x = a;//x最终会是根节点元素
    //先找到根节点
    while(father[x] != x) x = father[x];
    //反向把经过节点的父节点直接变为根节点，压缩路径
    while(father[a] != a){
        int b = a;
        a = father[a];
        father[b] = x;
    }
    return a;
}
//合并两个集合
void myunion(int a, int b){
    int af, bf;
    af = findFather(a);
    bf = findFather(b);
    //将一个集合的根节点指向另一个集合的根节点
    if(af != bf) father[bf] = af;
}


int main(){
    int n, m, a, b;
    scanf("%d%d", &n, &m);
    init(n);
    for(int i = 0; i < m; i++){
        scanf("%d%d", &a, &b);
        myunion(a, b);
    }
    for(int  i = 1; i <= n; i ++){
            //如果当前节点所在集合的根节点不是自身
            if(father[i] != i){
                isRoot[father[i]]++;
                isRoot[i] --;
            }
    }
    for(int i = 1; i <= n; i ++) printf("%d ", isRoot[i]);
    for(int i = 1; i <= n; i ++) printf("%d ", father[i]);
    system("pause");
    return 0;
}