剑指offer | 12. 矩阵中的路径

题目内容

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的3×4的矩阵中包含一条字符串“bfce”的路径（路径中的字母用加粗标出）。

[["a","b","c","e"],
["s","f","c","s"],
["a","d","e","e"]]

但矩阵中不包含字符串“abfb”的路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入这个格子。

示例 1：

输入：board = 
[["A","B","C","E"],
["S","F","C","S"],
["A","D","E","E"]], 
word = "ABCCED"
输出：true

示例 2：

输入：board = 
[["a","b"],
["c","d"]], 
word = "abcd"
输出：false

提示：

   1 <= board.length <= 200
   1 <= board[i].length <= 200

思路：

• 第一反应肯定是DFS，但是由于自己对dfs思想理解得还不是很透彻所以还是迟迟写不出来，看题解发现正确思路应该是dfs回溯+剪枝。
• 终止条件：
  • 1 返回false：
    • 行或列越界
    • 当前矩阵元素与目标字符不同（包括当前元素已被访问过后，修改为空字符串了。）
  • 2 返回true：
    • 字符串匹配完成

时间复杂度： $O(3^{K}MN)$;其中$3$表示每个位置往后走的时候，有上下左右四个方向，但是要去掉之前来的一个方向，$K$指的是需要匹配的字符串的长度$MN$指的是一共有MN个起点。

空间复杂度： $O(K)$;因为搜索过程中递归深度不超过K，所以每次系统调用累计使用的栈空间占用O（K），最坏情况下K=MN。

完整代码：

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        rows = board.size();
        cols = board[0].size();
        for(int i = 0; i < rows; i++) {
            for(int j = 0; j < cols; j++) {
                if(dfs(board, word, i, j, 0)) return true;
            }
        }
        return false;
    }
private:
    int rows, cols;
    bool dfs(vector<vector<char>>& board, string word, int i, int j, int k) {
        if(i >= rows || i < 0 || j >= cols || j < 0 || board[i][j] != word[k]) return false;
        if(k == word.size() - 1) return true;
        board[i][j] = '\0';
        bool res = dfs(board, word, i + 1, j, k + 1) || dfs(board, word, i - 1, j, k + 1) || 
                      dfs(board, word, i, j + 1, k + 1) || dfs(board, word, i , j - 1, k + 1);
        board[i][j] = word[k];
        return res;
    }
};

参考题解：面试题12. 矩阵中的路径（ DFS + 剪枝 ，清晰图解）