[DFS BFS] 200. 岛屿数量(DFS + used数组 + 空间优化)
200. 岛屿数量
题目链接:https://leetcode-cn.com/problems/number-of-islands/
相似题目:130. 被围绕的区域
分类:深度优先遍历、广度优先遍历、图、并查集(待补充)
思路1:DFS + 辅助数组 -> 空间优化
设置一个辅助二维数组记录每个元素是否被访问过。
算法运行的每一轮是在寻找一个岛屿,如果在寻找过程中发现某一元素被访问过,说明这个元素已经被记录为其他岛屿了,不能作为当前岛屿的节点。
关键问题:如何寻找当前岛屿的所有节点?
可以用递归回溯来实现。
每个节点都枚举自己上下左右相邻的节点,如果节点值为1且未被访问过,则置为访问过,然后以该节点为起点,重新判断它的上下左右。
如果上下左右全为0或已访问过的1,则直接return.
算法流程
- 设置一个辅助二维数组来记录节点是否被访问过,count计数器用来记录岛屿数量。
- 遍历grid数组,寻找每个岛屿的起点,如果某个节点=='1’且未被访问过,就可以作为一个新岛屿的起点,count+1,然后以该点为起点调用递归函数findIsland将同在一个岛屿上的其他节点全部置为“访问过”。
- 最终count存放的就是岛屿数量。
实现代码
class Solution {
public int numIslands(char[][] grid) {
//特殊用例:null , [], [[]]
if(grid == null || grid.length == 0 || grid[0].length == 0) return 0;
int rows = grid.length, cols = grid[0].length;
boolean[][] used = new boolean[rows][cols];
int count = 0;//记录岛屿的数量
//遍历整个grid数组,寻找岛屿的起点
for(int i = 0; i < rows; i++){
for(int j = 0; j < cols; j++){
//找到一个起点
if(grid[i][j] == '1' && !used[i][j]){
count++;//岛屿数量加1
findIsland(i, j, grid, used);//访问所有同一岛屿上的节点
}
}
}
return count;
}
//以一个节点为起点,寻找所在岛屿的所有节点
public void findIsland(int x, int y, char[][] grid, boolean[][] used){
if(grid[x][y] == '1' && !used[x][y])
used[x][y] = true;
else return;
//上
if(y - 1 >= 0) findIsland(x, y - 1, grid, used);
//下
if(y + 1 < grid[0].length) findIsland(x, y + 1, grid, used);
//左
if(x - 1 >= 0) findIsland(x - 1, y, grid, used);
//右
if(x + 1 < grid.length) findIsland(x + 1, y, grid, used);
}
}
- 时间复杂度:O((MN)^2),
- 空间复杂度:O(MN)
空间优化:可以借用grid本身作为used记录节点是否被访问过,因为grid的元素只有两种取值’0’或’1’,所以可以设置’2’表示该节点被访问过,即可不使用used辅助数组,将空间复杂度降为O(1).
- 当然能做这个优化的前提是允许修改输入数据。
实现代码:
class Solution {
public int numIslands(char[][] grid) {
//特殊用例:null , [], [[]]
if(grid == null || grid.length == 0 || grid[0].length == 0) return 0;
int rows = grid.length, cols = grid[0].length;
int count = 0;//记录岛屿的数量
//遍历整个grid数组,寻找岛屿的起点
for(int i = 0; i < rows; i++){
for(int j = 0; j < cols; j++){
//找到一个起点
if(grid[i][j] == '1'){
count++;//岛屿数量加1
findIsland(i, j, grid);//访问所有同一岛屿上的节点
}
}
}
return count;
}
//以一个节点为起点,寻找所在岛屿的所有节点
public void findIsland(int x, int y, char[][] grid){
if(grid[x][y] == '1')
grid[x][y] = '2';
else return;
//上
if(y - 1 >= 0) findIsland(x, y - 1, grid);
//下
if(y + 1 < grid[0].length) findIsland(x, y + 1, grid);
//左
if(x - 1 >= 0) findIsland(x - 1, y, grid);
//右
if(x + 1 < grid.length) findIsland(x + 1, y, grid);
}
}