LeetCode 79. 单词搜索

Description

给定一个二维网格和一个单词，找出该单词是否存在于网格中。

单词必须按照字母顺序，通过相邻的单元格内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

示例:

board =
[
  ['A','B','C','E'],
  ['S','F','C','S'],
  ['A','D','E','E']
]

给定 word = "ABCCED", 返回 true
给定 word = "SEE", 返回 true
给定 word = "ABCB", 返回 false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/word-search
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        row = len(board)
        col = len(board[0])
        self.res = False
        def backtrack(x, y, track, k):
            if self.res: return
            if len(track) == len(word):
                self.res = True
                return
            for i, j in [(-1, 0), (1, 0), (0, 1), (0, -1)]:
                x_ = x + i
                y_ = y + j
                if 0 <= x_ < row and 0 <= y_ < col and (x_, y_) not in visited and board[x_][y_] == word[k]:
                    visited.add((x_, y_))
                    track.append(board[x_][y_]) # 加上track方便调试
                    backtrack(x_, y_, track, k+1)
                    visited.remove((x_, y_))
                    track.pop()
        for i in range(row):
            for j in range(col):
                if board[i][j] == word[0]:
                    visited = set()
                    visited.add((i, j))
                    backtrack(i, j, [word[0]], 1)
                    if self.res: return True
        return False