leetcode-矩阵中的路径

 题目来自LeetCode，链接：面试题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

 这种题目很明显就是DFS的题。这里每次DFS搜索的下一步可以向四个方向，注意这里的剪枝是一旦发现一条合法路径就立即结束搜索，具体实现的时候需要对四个方向的返回结果进行或运算（这样可以保证剪枝成功，一旦有一个返回了true，后面的方向就不用遍历了）。一次DFS最多需要遍历整个矩阵，也就是最差情况下复杂度为$O(mn)$，因为最差情况下需要将所有位置作为起始点进行DFS，所以最差的总时间复杂度为$O(m^2{n}^2)$，空间复杂度为$O(mn)$。

 JAVA版代码如下：

class Solution {
    private boolean[][] visited;
    private char[] words;
    private int row;
    private int col;

    private boolean dfs(char[][] board, int i, int j, int targetIdx) {
        if (i < 0 || j < 0 || i >= row || j >= col || visited[i][j] || board[i][j] != words[targetIdx]) {
            return false;
        }
        if (targetIdx == words.length - 1) {
            return true;
        }
        else {
            // 标记访问
            visited[i][j] = true;
            // 注意这里四个要并列，以实现剪枝（一旦有一个返回true后面的就不用执行了）
            boolean find = dfs(board, i, j - 1, targetIdx + 1) || dfs(board, i, j + 1, targetIdx + 1)
                || dfs(board, i - 1, j, targetIdx + 1) || dfs(board, i + 1, j, targetIdx + 1);
            // 回溯，标记为未访问
            visited[i][j] = false;
            return find;
        }

    }
    public boolean exist(char[][] board, String word) {
        words = word.toCharArray();
        row = board.length;
        col = board[0].length;
        if (words.length > row * col) {
            return false;
        }
        visited = new boolean[row][col];
        boolean result = false;
        for (int i = 0; i < row; ++i) {
            for (int j = 0; j < col; ++j) {
                if (dfs(board, i, j, 0)) {
                    return true;
                }
            }
        }
        return false;
    }
}

 提交结果如下:

 Python版代码如下：

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        row = len(board)
        col = len(board[0])
        if row * col < len(word):
            return False
        visited = [[False for _ in range(col)] for _ in range(row)]

        def dfs(i, j, targetIdx):
            if i < 0 or j < 0 or i >= row or j >= col or visited[i][j] or board[i][j] != word[targetIdx]:
                return False
            if targetIdx == len(word) - 1:
                return True
            visited[i][j] = True
            find = dfs(i, j - 1, targetIdx + 1) or dfs(i, j + 1, targetIdx + 1) or dfs(i - 1, j, targetIdx + 1) or dfs(i + 1, j, targetIdx + 1)
            visited[i][j] = False
            return find

        for i in range(row):
            for j in range(col):
                if dfs(i, j, 0):
                    return True
        return False

 提交结果如下: