描述
这是一篇针对初学者的题解,给出一个比较好的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