刷题记录：单词搜索

单词搜索

class Solution:
    def exist(self, board, word):
        directions = [[-1, 0], [1, 0], [0, -1], [0, 1]]
        row, col = len(board), len(board[0])
        matrix = [[True] * col for _ in range(row)]
        ans = False
		# 从 i,j 位置开始递归搜索 word 第 idx 个字符
        def backtracking(idx, i, j):
            if idx == len(word): # 搜索成功返回结果
                nonlocal ans
                ans = True
                return
            for direction in directions: # 四个方向分别进行递归搜索
                if 0 <= i+direction[0] < row and 0 <= j+direction[1] < col:
                # 1）下标不能越界
                    if board[i+direction[0]][j+direction[1]] == word[idx] \
                    # 2）字母成功匹配
                            and (matrix[i+direction[0]][j+direction[1]] == True):
                            # 3） 字母未被重复利用    三个条件缺一不可
                        matrix[i+direction[0]][j+direction[1]] = False
                        # 标记字母为 “已经使用”
                        backtracking(idx+1, i+direction[0], j+direction[1])
                        # 从当前位置继续递归搜索
                        matrix[i + direction[0]][j + direction[1]] = True
                        # 回溯后 状态复原
        for i in range(row):
            for j in range(col):
                if board[i][j] == word[0]:
                # 从匹配了第一个字母的位置开始递归搜索
                    matrix[i][j] = False
                    # 标记字母为 “已经使用”
                    backtracking(1, i, j)
                    # 递归搜索
                    matrix[i][j] = True
                    # 搜索失败 状态复原
        return ans