面试题12 矩阵中的路径(路径搜索) Python3

请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的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

 

解题思路

采用深度有限搜索。从一个元素开始,按照上下左右的顺序进行深度优先遍历,成功则返回True;不成功则回溯至上一个节点。

剪枝。当当前路径无法匹配成功时(如元素不同或该元素已经被访问),则立刻返回,回溯至上一个节点。

 

 

代码

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        self.board = board
        self.word = word
        for i in range(len(board)): # 按行
            for j in range(len(board[0])):
                if self.dfs(i,j,0):
                    return True
        return False    # 遍历完后仍不能返回True,则返回False

    def dfs(self, i,j,k): # 第i行第j列,是否匹配第k个字符
        board = self.board
        word = self.word
        if not (0 <= i < len(board)) or not (0 <= j < len(board[0])) or (board[i][j] != word[k]): # 不匹配的情况
            return False
        # 匹配的情况
        if k == len(word) - 1:  # 如果k是最后一个字符,则返回True
            return True
        tmp, board[i][j] = board[i][j], '/' # 如果不是最后一个字符,有必要将这个字符暂时置为'/',以防止重复
        res = self.dfs(i+1,j,k+1) or self.dfs(i-1,j, k+1) or self.dfs(i, j+1, k+1) or self.dfs(i, j-1, k+1)  # 对上下左右依次进行搜索
        board[i][j] = tmp   # 搜索完成后,将board[i][j]的值复原
        return res

 

总结

1. Python可以使用 <=i<=这样的判断语句;

2. 二维list a和a[0]的含义.

3. 类中的函数f调用自身递归需要加上self.f

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章