算法分类（三）并查集

一、概念理解

1.初始化，最开始每个元素是单独的，自己的父亲指向自己

const n = 10
let fa = new Array(n)
for (let i = 1; i <= n; i++) {
  fa[i] = i
}

2.合并，让2个元素指向同一个root

这里有2个重点，一个是查询根节点，一个是合并

查询根节点：

const find = (x: number): number => {
  if (fa[x] != x) {
    //这一步，能够实现路径压缩，让所有元素指向root
    fa[x] = find(fa[x])
  }
  return fa[x]
}

合并：

const union = (p: number, q: number) => {
  const faP = find(p)
  const faQ = find(q)
  fa[faP] = faQ
}

二、例题

547. 省份数量

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中 省份 的数量。

function findCircleNum(isConnected: number[][]): number {
    //初始化
    const n = isConnected.length
    let count = n //count代表没有相连的城市，最开始n
    let fa = new Array(n)
    for (let i = 0; i < n; i++) {
        fa[i] = i
    }

    //find
    const find = (x: number): number => {
        if (fa[x] != x) {
            //这一步，能够实现路径压缩，让所有元素指向root
            fa[x] = find(fa[x])
        }
        return fa[x]
    }

    //union
    const union = (p: number, q: number) => {
        const faP = find(p)
        const faQ = find(q)
        fa[faP] = faQ
    }

    //把每个相连的城市，进行连接
    for (let i = 0; i < n; i++) {
        for (let j = i + 1; j < n; j++) {
            if (isConnected[i][j] === 1) {
                const faI = find(i)
                const faJ = find(j)
                if (faI != faJ) {
                    fa[faI] = faJ
                    count--
                }
            }
        }
    }

    return count
};