[路飞]_200. 岛屿数量

110 阅读1分钟

题目介绍

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

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

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

 

示例 1:

输入:grid = [ ["1","1","1","1","0"], ["1","1","0","1","0"], ["1","1","0","0","0"], ["0","0","0","0","0"] ] 输出:1 示例 2:

输入:grid = [ ["1","1","0","0","0"], ["1","1","0","0","0"], ["0","0","1","0","0"], ["0","0","0","1","1"] ] 输出:3  

提示:

m == grid.length n == grid[i].length 1 <= m, n <= 300 grid[i][j] 的值为 '0' 或 '1'

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/nu… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

并查集的思路,需要值为1的上下左右有节点连接,其实可以转化为 每个节点的需要和上节点和左节点连接,然后找到节点值为1的集合数

代码

/**
 * @param {character[][]} grid
 * @return {number}
 */
class UnionSet {
  constructor(n) {
    this.fa = new Array(n + 1)
    for (let i = 0; i <= n; i++) {
      this.fa[i] = i
    }
  }
  get(x) {
    let fa = this.fa
    return fa[x] = (fa[x] === x ? x : this.get(fa[x]))
  }
  merge(a, b) {
    this.fa[this.get(a)] = this.get(b)
  }
}
var numIslands = function(grid) {
    let n = grid.length
    let m = grid[0].length
    function findInd(i, j) {
        return i * m + j
    }
    let unionSet = new UnionSet(n * m)

    for (let i = 0; i < n; i++) {
        for (let j = 0; j < m; j++) {
            if (grid[i][j] === '0') continue
            if (i > 0 && grid[i - 1][j] === '1') {
                unionSet.merge(findInd(i, j), findInd(i - 1, j))
            }
            if (j > 0 && grid[i][j - 1] === '1') {
                unionSet.merge(findInd(i, j), findInd(i, j - 1))
            }
        }
    }
    let ans = 0
    for (let i = 0; i < n; i++) {
        for (let j = 0; j < m; j++) {
            if (grid[i][j] === '1' && unionSet.get(findInd(i, j)) === findInd(i, j)) ans += 1
        }
    }
    return ans
};