Skip to content

岛屿数量

Posted on:2024年7月20日 at 10:37

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

提示:

/**
 * @param {character[][]} grid
 * @return {number}
 */
var numIslands = function (grid) {};

方法一:深度优先搜索

我们可以将二维网格看成一个无向图,竖直或水平相邻的 11 之间有边相连。

为了求出岛屿的数量,我们可以扫描整个二维网格。如果一个位置为 11,则以其为起始节点开始进行深度优先搜索。在深度优先搜索的过程中,每个搜索到的 11 都会被重新标记为 00。

最终岛屿的数量就是我们进行深度优先搜索的次数。

const numIslands = (grid) => {
  let count = 0;
  for (let i = 0; i < grid.length; i++) {
    for (let j = 0; j < grid[0].length; j++) {
      if (grid[i][j] === "1") {
        count++;
        turnZero(i, j, grid);
      }
    }
  }
  return count;
};
function turnZero(i, j, grid) {
  if (
    i < 0 ||
    i >= grid.length ||
    j < 0 ||
    j >= grid[0].length ||
    grid[i][j] === "0"
  )
    return;
  grid[i][j] = "0";
  turnZero(i, j + 1, grid);
  turnZero(i, j - 1, grid);
  turnZero(i + 1, j, grid);
  turnZero(i - 1, j, grid);
}

复杂度分析

方法二:广度优先搜索

同样地,我们也可以使用广度优先搜索代替深度优先搜索。

为了求出岛屿的数量,我们可以扫描整个二维网格。如果一个位置为 11,则将其加入队列,开始进行广度优先搜索。在广度优先搜索的过程中,每个搜索到的 11 都会被重新标记为 00。直到队列为空,搜索结束。

最终岛屿的数量就是我们进行广度优先搜索的次数。

const numIslands = (grid) => {
  let count = 0;
  let queue = [];
  for (let i = 0; i < grid.length; i++) {
    for (let j = 0; j < grid[0].length; j++) {
      if (grid[i][j] === "1") {
        count++;
        grid[i][j] = "0"; // 做标记,避免重复遍历
        queue.push([i, j]);
        turnZero(queue, grid);
      }
    }
  }
  return count;
};
function turnZero(queue, grid) {
  const dirs = [
    [0, 1],
    [1, 0],
    [0, -1],
    [-1, 0],
  ];
  while (queue.length) {
    const cur = queue.shift();
    for (const dir of dirs) {
      const x = cur[0] + dir[0];
      const y = cur[1] + dir[1];
      if (
        x < 0 ||
        x >= grid.length ||
        y < 0 ||
        y >= grid[0].length ||
        grid[x][y] !== "1"
      ) {
        continue;
      }
      grid[x][y] = "0";
      queue.push([x, y]);
    }
  }
}

复杂度分析

方法三:并查集

同样地,我们也可以使用并查集代替搜索。

为了求出岛屿的数量,我们可以扫描整个二维网格。如果一个位置为 1,则将其与相邻四个方向上的 1 在并查集中进行合并。

最终岛屿的数量就是并查集中连通分量的数目。

/**
 * @param {character[][]} grid
 * @return {number}
 */
var numIslands = function (grid) {
  const Y = grid.length;
  const X = grid[0].length;
  const uf = new UnionFind();

  for (let i = 0; i < Y; i++) {
    for (let j = 0; j < X; j++) {
      if (grid[i][j] == 1) uf.makeSet([i, j]);
    }
  }

  for (let i = 0; i < Y; i++) {
    for (let j = 0; j < X; j++) {
      if (grid[i][j] == 1) {
        console.log(i, j);
        if (i + 1 < Y && grid[i + 1][j] == 1) uf.union([i, j], [i + 1, j]); // 右侧
        if (j + 1 < X && grid[i][j + 1] == 1) uf.union([i, j], [i, j + 1]); // 下侧
      }
    }
  }

  return uf.getCount();
};
class UnionFind {
  constructor() {
    this.parents = {};
    this.count = 0;
  }
  makeSet(x) {
    this.parents[x] = x + "";
    this.count++;
  }
  findSet(x) {
    // 路径压缩,查x的根节点
    while (this.parents[x] !== x + "") {
      x = this.parents[x];
    }
    return x + "";
  }
  union(x, y) {
    this.link(this.findSet(x), this.findSet(y));
  }
  link(x, y) {
    if (x === y) return;
    this.parents[x] = y;
    this.count--;
  }
  getCount() {
    return this.count;
  }
}

复杂度分析

原文转自:https://fe.ecool.fun/topic/aa2840ae-f884-4fcb-a08b-e20310654dbd