[路飞]_leetcode-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

提示：

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

解题思路

本题是一个连通性问题。
因为陆地相连组成岛屿，所以我们可以通过如下方式获取岛屿数量。

1. 遍历输入二维数组，如果当前值为 1,则说明是一块陆地，而陆地必然属于岛屿，所以更新岛屿数量+1。
2. 将当前陆地修改为 0,又因为相邻的陆地都是属于同一个岛屿，所以找到相邻的陆地，同样修改为 0。
3. 经过步骤2处理，就可以将每一个岛屿数量获取到，并将其对应陆地修改为 0,随意后续遍历到值为 1 的元素，必然属于一个新的岛屿，此时重复步骤2即可。
4. 最后遍历完输入数组，就得到了岛屿的数量。

代码实现

// 先搜索二维数组，如果一个位置为1，判断当前节点的上下左右位置是否为1
// 是的话开始合并，最后出来的连通分量就是要求的岛屿数量;
let count;
var numIslands = function(grid) {
    let m = grid.length;
    let n = grid[0].length;
    let uf = new UnionFind(m * n);
    // 1.对二维数组进行搜索
    count = 0;
    for(let i = 0; i < m ; i++){
        for(let j = 0; j < n; j++){
            if(grid[i][j] == '1'){
                count ++;
            }
        }
    }
    // 2.对当前节点上下左右，四个方向的值进行判断
    for(let i = 0; i < m; i++){
        for(let j = 0 ; j < n; j++){
            if(grid[i][j] == '1'){ // 岛屿数量+1；
                if(i -1 >= 0 && grid[i-1][j] == '1'){//当前节点上一个节点
                    uf.unite(i * n + j, (i - 1)* n + j)
                }
                if(i + 1 < m  && grid[i+1][j] == '1'){//当前节点下一个节点
                    uf.unite(i * n + j, (i + 1)* n + j)
                }
                if(j - 1 >=0  && grid[i][j-1] == '1'){//当前节点左一个节点
                    uf.unite(i * n + j, i * n + j - 1)
                }
                if(j + 1 < n  && grid[i][j+1] == '1'){//当前节点右一个节点
                    uf.unite(i * n + j,  i * n + j +1)
                }
            }
        }
    }
    return count; // 连通分量就是我们的岛屿数量
};
class UnionFind{
    constructor(n){
        this.parent = new Array(n).fill(0).map((v,index)=>index);
        this.rank = new Array(n).fill(1);
        this.setCount = n;
    }
    finSet(index){
        if(this.parent[index] != index){
            this.parent[index] = this.finSet(this.parent[index])
        }
        return this.parent[index]
    }
    unite(index1,index2){
        let root1 = this.finSet(index1),root2 = this.finSet(index2);
        if( root1 != root2){
            if(this.rank[root1] < this.rank[root2]){
                [root1,root2] = [root2,root1]
            }
            this.parent[root2] = root1;
            this.rank[root1] += this.rank[root1];
            this.setCount --;
            count --;
        }
    }
    getCount(){
        return this.setCount;
    }
    connected(index1,index2){
        let root1 = this.finSet(index1),root2 = this.finSet(index2)
        return root1 == root2;;
    }
}

至此我们就完成了leetcode-200-岛屿数量