DFS与回溯的区别——以力扣《694.不同的岛屿数量》为例

dotJunz

已于 2022-12-19 09:55:26 修改

阅读量157

点赞数

分类专栏：算法 Java 文章标签：深度优先 leetcode 算法

于 2022-12-19 09:45:58 首次发布

本文链接：https://blog.csdn.net/m0_46283220/article/details/128368709

版权

算法同时被 2 个专栏收录

83 篇文章 1 订阅

订阅专栏

Java

31 篇文章 0 订阅

订阅专栏

前言

对于最后一题不同的岛屿数量，最后提到这里涉及DFS和回溯算法的一个细微差别，下面谈一谈我的理解。

题目

694.不同的岛屿数量

比如题目输入下面这个二维矩阵：

在这里插入图片描述
其中有四个岛屿，但是左下角和右上角的岛屿形状相同，所以不同的岛屿共有三个，算法返回 3。

DFS与回溯的区别

本题要用到序列化，遍历顺序如下图所示
在这里插入图片描述

DFS

我认为东哥的做法是DFS，关注在节点
东哥做法：

DFS函数

void dfs(int[][] grid, int i, int j, StringBuilder sb, int dir) {
    int m = grid.length, n = grid[0].length;
    if (i < 0 || j < 0 || i >= m || j >= n 
        || grid[i][j] == 0) {
        return;
    }
    // 前序遍历位置：进入 (i, j)
    grid[i][j] = 0;
    sb.append(dir).append(',');
    
    dfs(grid, i - 1, j, sb, 1); // 上
    dfs(grid, i + 1, j, sb, 2); // 下
    dfs(grid, i, j - 1, sb, 3); // 左
    dfs(grid, i, j + 1, sb, 4); // 右
    
    // 后序遍历位置：离开 (i, j)
    sb.append(-dir).append(',');
}

主函数

int numDistinctIslands(int[][] grid) {
    int m = grid.length, n = grid[0].length;
    // 记录所有岛屿的序列化结果
    HashSet<String> islands = new HashSet<>();
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (grid[i][j] == 1) {
                // 淹掉这个岛屿，同时存储岛屿的序列化结果
                StringBuilder sb = new StringBuilder();
                // 初始的方向可以随便写，不影响正确性
                dfs(grid, i, j, sb, 666);
                islands.add(sb.toString());
            }
        }
    }
    // 不相同的岛屿数量
    return islands.size();
}

回溯

回溯函数

int[][] dir = new int[][]{{0, 1}, {1, 0}, {0, -1}, {-1, 0}};

void dfs(int[][] grid, int i, int j, StringBuilder sb) {
    int m = grid.length, n = grid[0].length;
    if (i < 0 || j < 0 || i >= m || j >= n
        || grid[i][j] == 0) {
        return;
    }
    grid[i][j] = 0;

    for(int k = 1; k <= 4; k++){
        int ni = i + dir[k - 1][0];
        int nj = j + dir[k - 1][1];
        // 前序遍历位置：进入 (ni, nj)
        sb.append(k).append(',');
        dfs(grid, ni, nj, sb);
        // 后序遍历位置：离开 (ni, nj)
        sb.append(-k).append(',');
    }
}