0%

算法系列-深度优先搜索dfs

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
class Solution {
private:
bool inArea(vector<vector<char>>& grid,int i, int j) {
return i >= 0 && j >= 0 && i < grid.size() && j < grid[0].size();//判断坐标是否合法
}
void dfs(vector<vector<char>>& grid,int i, int j) {
int row = grid.size();
int col = grid[0].size();
if(!inArea(grid, i, j)) return;//先把数值读进来,再判断是不是符合规则,当然直接放在下面判断也行,但读起来有点冗余
if(grid[i][j] != '1') return;
grid[i][j] = '2';
dfs(grid, i - 1, j);
dfs(grid, i + 1, j);
dfs(grid, i, j - 1);
dfs(grid, i, j + 1);
}
public:
int numIslands(vector<vector<char>>& grid) {
int row = grid.size();
int col = grid[0].size();
int count = 0;
for(int i = 0;i < row;i++) {
for(int j = 0;j < col;j++) {
if(grid[i][j] == '1') {
dfs(grid, i, j);
count++;
}
}
}
return count;
}
};

首先,dfs的逻辑应该包括判断base case,即在知道目前输入的情况下,应该怎么处理当下的值;

其次,应该思考下一步状态应该往那个方向转移,状态的表征是通过改变dfs的传入参数来实现的。