js并查集算法

// 547 medium 朋友圈

// 班上有 N 名学生。其中有些人是朋友，有些则不是。他们的友谊具有是传递性。
// 如果已知 A 是 B 的朋友，B 是 C 的朋友，那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈，是指所有朋友的集合。
// 给定一个 N * N 的矩阵 M，表示班级中学生之间的朋友关系。如果Mi = 1，
// 表示已知第 i 个和 j 个学生互为朋友关系，否则为不知道。你必须输出所有学生中的已知的朋友圈总数。
//
// 示例 1:
// 输入:
// [[1,1,0],
// [1,1,0],
// [0,0,1]]
// 输出: 2
// 说明：已知学生0和学生1互为朋友，他们在一个朋友圈。
// 第2个学生自己在一个朋友圈。所以返回2。

// 示例 2:
// 输入:
// [[1,1,0],
// [1,1,1],
// [0,1,1]]
// 输出: 1
// 说明：已知学生0和学生1互为朋友，学生1和学生2互为朋友，所以学生0和学生2也是朋友，所以他们三个在一个朋友圈，返回1。

// 注意：
// N 在[1,200]的范围内。
// 对于所有学生，有Mi = 1。
// 如果有Mi = 1，则有Mj = 1。
//

// 并查集类。需要维护一个数组和两个方法，find()和union()
class bingchaSet{
    constructor(n) {
        this.arr = []
        // 初始默认为每个人是独立圈子，则他的父级就是他自身
        for(let i = 0; i < n; i++){
            this.arr[i] = i;
        }
    }

    // 直到arr[x]==x，停止向上搜索
    find(x){
        let arr = this.arr;
        while(arr[x] !== x){
            x = arr[x];
        }
        return arr[x]
    }

    // 路径压缩
    union(x, y){
        let xFather = this.find(x)
        let yFather = this.find(y)
        if(xFather!==yFather){
            this.arr[xFather] = yFather;
        }
    }
}

/**
 * @param {number[][]} M
 * @return {number}
 */
var findCircleNum = function(M) {

    let N = M.length;
    let set = new bingchaSet(N)
    for(let i = 0; i < N; i++){
        for(let j = i+1; j < N; j++){
            if(M[i][j] === 1){
                set.union(j, i)
            }
        }
    }


    let count = 0;
    for(let i = 0; i < set.arr.length; i++){
        if(set.arr[i] === i){
            count++;
        }
    }
    return count;
};

console.log(findCircleNum([[1,0,0,1],[0,1,1,0],[0,1,1,1],[1,0,1,1]]))

https://github.com/cunzaizhuy...
我的前端算法库