[路飞] 岛屿数量

题目描述

给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

解题思路

数据结构

并查集

过程

二维数组行数为 m，列数 n

将所有的网格（二位数组元素）在并查集的 fa 中进行一个编号

遍历二维数组，这个过程中，要进行关键一步：

如果一个元素的左或上，以及元素本身是 '1'，那么我们需要执行 merge 操作，index 是 i * m + j

最后生成完并查集之后，我们计算结果

仍然要遍历每个二维数组的元素，如果在二维数组中，该元素为 '1'，并且在并查集中，该元素的 root 就是他本身，那么我们认为，这个元素是一个单独的岛屿

代码

/**
 * @param {character[][]} grid
 * @return {number}
 */
var numIslands = function (grid) {
  const n = grid.length
  const m = grid[0].length
  let ans = 0

  const set = new UnionSetTemplate(m * n)

  for (let i = 0; i < n; i++) {
    for (let j = 0; j < m; j++) {
      if (grid[i][j] === "0") continue

      // if top item is 1
      if (i > 0 && grid[i - 1][j] === "1")
        set.merge(getIndex(i - 1, j), getIndex(i, j))
      // if left item is 1
      if (j > 0 && grid[i][j - 1] === "1") {
        set.merge(getIndex(i, j - 1), getIndex(i, j))
      }
    }
  }

  for (let i = 0; i < n; i++) {
    for (let j = 0; j < m; j++) {
      if (grid[i][j] === "1" && set.get(getIndex(i, j)) === getIndex(i, j))
        ans++
    }
  }
  return ans

  function getIndex(i, j) {
    return i * m + j
  }
}

class UnionSetTemplate {
  constructor(n) {
    this.fa = []
    for (let i = 0; i <= n; i++) {
      this.fa[i] = i
    }
  }

  get(x) {
    return (this.fa[x] = this.fa[x] === x ? x : this.get(this.fa[x]))
  }

  merge(a, b) {
    this.fa[this.get(a)] = this.get(b)
  }
}