题目介绍
给你一个由 '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
};