目录
DFS递归核心代码:
void dfs(char[][] grid, int i, int j) {
//四个方向深度搜索
dfs(grid, i + 1, j);
dfs(grid, i, j + 1);
dfs(grid, i - 1, j);
dfs(grid, i, j - 1);
}
1.先看一个力扣题,记一下模版代码,大部分DFS都是这样的:
200.岛屿的数量:
思路:如果有5个岛屿是相邻的,且这5个岛屿周围都是水,那么这5个岛屿看作1个岛屿
我采用的方法是只要这一个地方是岛屿,那么开始DFS,把周围相邻的所有岛屿变成水域,退出递归后数量加1.
class Solution {
// 主函数,计算岛屿数量
public int numIslands(char[][] grid) {
int res = 0;
int m = grid.length, n = grid[0].length;
// 遍历 grid
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == '1') {
// 每发现一个岛屿,岛屿数量加一
res++;
// 然后使用 DFS 将岛屿淹了
dfs(grid, i, j);
}
}
}
return res;
}
// 从 (i, j) 开始,将与之相邻的陆地都变成海水
void dfs(char[][] grid, int i, int j) {
int m = grid.length, n = grid[0].length;
if (i < 0 || j < 0 || i >= m || j >= n) {
// 超出索引边界
return;
}
if (grid[i][j] == '0') {
// 已经是海水了
return;
}
// 将 (i, j) 变成海水
grid[i][j] = '0';
// 淹没上下左右的陆地
dfs(grid, i + 1, j);
dfs(grid, i, j + 1);
dfs(grid, i - 1, j);
dfs(grid, i, j - 1);
}
}
代码很简单,无非是2层for循环+递归
2.把这个代码稍作修改就能用作其他的DFS题目了:
695.岛屿的最大面积:
class Solution {
int max=0,Max=0;
public int maxAreaOfIsland(int[][] grid) {
int k=0;
int m = grid.length, n = grid[0].length;
// 遍历 grid
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 1) {
// 然后使用 DFS 将岛屿淹了
max=0;
dfs(grid, i, j);
if(max>Max)Max=max;
}
}
}
return Max;
}
// 从 (i, j) 开始,将与之相邻的陆地都变成海水
void dfs(int[][] grid, int i, int j) {
int m = grid.length, n = grid[0].length;
if (i < 0 || j < 0 || i >= m || j >= n) {
// 超出索引边界
return;
}
if (grid[i][j] == 0) {
// 已经是海水了
return;
}
// 将 (i, j) 变成海水
grid[i][j] = 0;
max++;
// 淹没上下左右的陆地
dfs(grid, i + 1, j);
dfs(grid, i, j + 1);
dfs(grid, i - 1, j);
dfs(grid, i, j - 1);
}
}
2658.网格图中鱼的最大数目:
class Solution {
int max=0,Max=0;
public int findMaxFish(int[][] grid) {
int m = grid.length, n = grid[0].length;
// 遍历 grid
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] >0) {
// 然后使用 DFS 将岛屿淹了
max=0;
dfs(grid, i, j);
if(max>Max)Max=max;
}
}
}
return Max;
}
// 从 (i, j) 开始,将与之相邻的陆地都变成海水
void dfs(int[][] grid, int i, int j) {
int m = grid.length, n = grid[0].length;
if (i < 0 || j < 0 || i >= m || j >= n) {
// 超出索引边界
return;
}
if (grid[i][j] == 0||grid[i][j] == -1) {
// 已经是海水了
return;
}
max+=grid[i][j];
// 将 (i, j) 变成海水
grid[i][j] = -1;
// 淹没上下左右的陆地
dfs(grid, i + 1, j);
dfs(grid, i, j + 1);
dfs(grid, i - 1, j);
dfs(grid, i, j - 1);
}
}