[路飞]_200. 岛屿数量

项目地址

题目描述

给你一个由 '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

解题思路

• 由题可知，岛屿的话，必须是上下左右都必须是0，这样就形成了一个单独的岛屿，要不然就会形成连接为陆地
• 所有此题的目的就是为了求有几个集合

代码

// 编写一个并查集模板
class UnionSet{
    constructor(n) {
        this.parent = new Array(n).fill(0).map((value, index) => index);
    }
    get(x) {
        return this.parent[x] = (this.parent[x] == x ? x : this.get(this.parent[x]));
    }
    merge(a, b) {
        this.parent[this.get(a)] = this.get(b);
    }
}
var numIslands = function(grid) {
    const n = grid.length;
    const m = grid[0].length;
    const uf = new UnionSet(m * n);
    
    // 两次遍历
    for(let i = 0; i < n; i++) {
        for(let j = 0; j < m; j++) {
            // 如果当前节点为0，那么就什么都不做
            if(grid[i][j] == 0) continue;
            // 我们只需要连接当前节点的上节点和左节点即可
            // 如果i大于0，且为1，就去查看当前节点的上面节点是否是1，如果同为1，那么就连接
            if(i > 0 && grid[i - 1][j] == 1) uf.merge(ind(i, j), ind(i - 1, j));
            // 如果j大于0，且为1，就去查看当前节点的左侧节点是否是1，如果同为1，那么就连接
            if(j > 0 && grid[i][j -1] == 1) uf.merge(ind(i, j), ind(i, j - 1))
        }
    }
    let ans = 0;
    for(let i = 0; i < n; i++) {
        for(let j = 0; j < n; j++) {
            // 是否是同一集合的特点是：当前节点为1，且当前节点的get值是本身
            // 具体看下图打印
            if(grid[i][j] == 1 && uf.get(ind(i, j)) == ind(i, j)) ans += 1;
        }
    }
    return ans;
    function ind(i, j) {
        return (i) * m + j
    }
}