JS算法-岛屿数量

156 阅读2分钟

题目

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

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

题解

首先,定义了一个 numIslands 函数,它接收一个二维矩阵作为参数,返回岛屿数量。初始化变量 res 为 0,表示岛屿数量为 0。定义一个标志变量 flag 为 true,表示还有未访问的陆地。进入 while 循环,当 flag 为 true 时,表示还有未访问的陆地,需要继续搜索,在 for 循环中,遍历二维矩阵中的每一个元素,如果当前元素是 '1',表示发现了一块未访问的陆地,将 flag 置为 true,并且岛屿数量 res 加 1。然后调用 changeGrid 函数,将当前位置的陆地标记为已访问,并继续搜索与当前陆地相连的所有陆地,changeGrid 函数的作用是将当前位置的陆地标记为已访问,并且继续搜索与当前陆地相连的所有陆地。首先,将当前位置的陆地标记为已访问,即将 grid[i][j] 的值从 '1' 改为 '0'。然后,查找当前位置相邻的未访问陆地,如果找到,则将其加入到 arr 数组中。最后,如果 arr 数组不为空,则遍历 arr 数组中的元素,递归调用 changeGrid 函数,继续搜索与当前陆地相连的所有陆地。如果 arr 数组为空,则表示当前陆地没有相邻的未访问陆地,搜索结束

var numIslands = function (grid) {
  let res = 0
  let flag = true
  while (flag) {
    flag = false
    for (let i = 0; i < grid.length; i++) {
      for (let j = 0; j < grid[0].length; j++) {
        if (grid[i][j] == '1') {
          flag = true
          res++
          changeGrid(i, j, grid)
          break
        }
      }
      if (flag) break
    }
  }
  return res
};
const changeGrid = (i, j, grid) => {
  grid[i].splice(j, 1, '0')
  let arr = []
  if (j + 1 < grid[0].length && grid[i][j + 1] == '1') {
    arr.push([i, j + 1])
  }
  if (j - 1 >= 0 && grid[i][j - 1] == '1') {
    arr.push([i, j - 1])
  }
  if (i + 1 < grid.length && grid[i + 1][j] == '1') {
    arr.push([i + 1, j])
  }
  if (i - 1 >= 0 && grid[i - 1][j] == '1') {
    arr.push([i - 1, j])
  }
  if (arr.length === 0) return
  for (let item of arr) {
    changeGrid(item[0], item[1], grid)
  }
}

坚持努力,无惧未来!