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
解题思路
思路一: DFS(深度优先搜索)
目标是找到矩阵中 “岛屿的数量” ,上下左右相连的 1 都被认为是连续岛屿。
dfs方法: 设目前指针指向一个岛屿中的某一点 (row, col),寻找包括此点的岛屿边界。
从 (row, col) 向此点的上下左右 (row + 1, col), (row - 1, col), (row, col + 1),(row, col - 1) 做深度搜索。
终止条件:
(row, col) 越过矩阵边界;
grid[row][col] === ‘0’,代表此分支已越过岛屿边界。
搜索岛屿的同时,执行 grid[row][col] = ‘0’(沉没
),即将岛屿所有节点删除,以免之后重复搜索相同岛屿。
主循环:
遍历整个矩阵,当遇到 grid[row][col] === ‘1’ 时,从此点开始做深度优先搜索 dfs,岛屿数 count + 1 且在深度优先搜索中删除此岛屿。
最终返回岛屿数 count 即可。
/**
* @param {character[][]} grid
* @return {number}
*/
var numIslands = function(grid) {
let count = 0, rowLen = grid.length, colLen = grid[0].length;
const dfs = function (row, col) {
if (row < 0 || row >= rowLen || col < 0 || col >= colLen || grid[row][col] === '0') {
return;
}
grid[row][col] = '0'; // 沉没
dfs(row, col - 1);
dfs(row - 1, col);
dfs(row + 1, col);
dfs(row, col + 1);
}
for (let row = 0; row < rowLen; row++) {
for (let col = 0; col < colLen; col++) {
if (grid[row][col] === '1') {
count++;
dfs(row, col);
}
}
}
return count;
};
-
时间复杂度:O(MN),其中 M 和 N 分别为行数和列数。
-
空间复杂度:O(MN),在最坏情况下,整个网格均为陆地,深度优先搜索的深度达到MN。
思路二: BFS(广度优先遍历)
bfs 方法:
- 借用一个队列 queue,判断队列首部节点 (row, col) 是否未越界且为 1:
- 若是则置零(删除岛屿节点),并将此节点上下左右节点 (row + 1, col), (row - 1, col), (row, col + 1),(row, col - 1) 加入队列;
- 若不是则跳过此节点;
- 循环 pop 队列首节点,直到整个队列为空,此时已经遍历完此岛屿。
实现代码如下:
/**
* @param {character[][]} grid
* @return {number}
*/
var numIslands = function(grid) {
let count = 0, rowLen = grid.length, colLen = grid[0].length;
const bfs = function (i, j) {
let queue = [];
queue.push({row: i, col: j});
while(queue.length) {
let {row, col} = queue.shift();
if (row >= 0 && row < rowLen && col >= 0 && col < colLen && grid[row][col] === '1') {
grid[row][col] = '0'; // 沉没
bfs(row, col - 1);
bfs(row - 1, col);
bfs(row + 1, col);
bfs(row, col + 1);
}
}
}
for (let row = 0; row < rowLen; row++) {
for (let col = 0; col < colLen; col++) {
if (grid[row][col] === '1') {
count++;
bfs(row, col);
}
}
}
return count;
};
-
时间复杂度:O(MN),其中 M 和 N 分别为行数和列数。
-
空间复杂度:O(min(M,N)),在最坏情况下,整个网格均为陆地,队列的大小可以达到 min(M,N)。
注意:
把「已遍历过的陆地格子」标记为和海洋格子一样的 0, 有很大隐患,因为这样我们就无法区分「海洋格子」和「已遍历过的陆地格子」了。如果题目更复杂一点,这很容易出 bug。