描述

这是一篇针对初学者的题解,给出一个比较好的DFS模板。
知识点:DFS
难度:二星


题解

题目描述:给定一个二维字符串矩阵mat,和一个字符串str,判断str是否可以在mat中匹配。
可以选择的方向是上下左右。

方法:DFS

这道题大家都知道是DFS的题,关键是怎么可以快速并且正确的写出,是本题解讨论的重点。
首先解释一下递归函数。
递归函数:就是当前处理的问题是什么,并且下一次在规模减小的情况下处理相同的问题。
比如此题:当前处理的问题是:判断字符串str[0 ... len]是否在mat中匹配,显然下一次递归处理的问题是:如果str[0]已经匹配,则判断字符串str[1 ... len]是否在mat中匹配。

这里先给出一个我认为比较清晰的DFS模板:

dfs(){

    // 第一步,检查下标是否满足条件

    // 第二步:检查是否被访问过,或者是否满足当前匹配条件

    // 第三步:检查是否满足返回结果条件

    // 第四步:都没有返回,说明应该进行下一步递归
    // 标记
    dfs(下一次)
    // 回溯
}  


main() {
    for (对所有可能情况) {
        dfs()
    }
}

上面每步的顺序都不能颠倒。

所以,对于这道题来说,首先dfs()的参数是什么,返回值是什么。
可以像这样:

// i, j 表示mat中的位置, pos表示当前正在匹配的字符串str的下标
// 成功返回整个字符串str, 则返回true, 否则返回false
bool dfs(int i, int j, int pos, char *str);

主函数该怎么写?可以像如下这样:

// 几个全局变量,便于程序的简洁,只是在刷题中建议
char *mat = 0;
int h = 0, w = 0;
int str_len = 0;
int dir[5] = {-1, 0, 1, 0, -1};
bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        mat = matrix;
        h = rows, w = cols;
         str_len = strlen(str);

        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (dfs(i, j, 0, str)) {
                    return true;
                }
            }
        }
        return false;
    }

然后套用模板,代码如下:

class Solution {
public:
    char *mat = 0;
    int h = 0, w = 0;
    int str_len = 0;
    int dir[5] = {-1, 0, 1, 0, -1};

    bool dfs(int i, int j, int pos, char *str) {
        // 因为dfs调用前,没有进行边界检查,
        // 所以需要第一步进行边界检查,
        // 因为后面需要访问mat中元素,不能越界访问
        if (i < 0 || i >= h || j < 0 || j >= w) {
            return false;
        }

        char ch = mat[i * w + j];
        // 判断是否访问过
        // 如果没有访问过,判断是否和字符串str[pos]匹配
        if (ch == '#' || ch != str[pos]) {
            return false;
        }

         // 如果匹配,判断是否匹配到最后一个字符
        if (pos + 1  == str_len) {
            return true;
        }

        // 说明当前字符成功匹配,标记一下,下次不能再次进入
        mat[i * w + j] = '#';
        for (int k = 0; k < 4; ++k) {
            if (dfs(i + dir[k], j + dir[k + 1], pos + 1, str)) {
                return true;
            }
        } 
        // 如果4个方向都无法匹配 str[pos + 1]
        // 则回溯, 将'#' 还原成 ch          
        mat[i * w + j] = ch;
        // 说明此次匹配是不成功的
        return false;   
    }
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        mat = matrix;
        h = rows, w = cols;
         str_len = strlen(str);

        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (dfs(i, j, 0, str)) {
                    return true;
                }
            }
        }
        return false;
    }
};

时间复杂度:O(3^k), 每个位置除当前自己的方向,还有3个方向可以展开。k为str的长度
空间复杂度:O(k), 最大递归栈的深度为k