leetcode-212-单词搜索②

题目描述：

第一次提交：（超出时间限制）

class Solution:
    def findWords(self, board: List[List[str]], words: List[str]) -> List[str]:
        def dfs(word,i,j,visited):
            if len(word)==1 and word==board[i][j]:
                return True
            elif word[0]!=board[i][j]:return
            else:
                list = [[i-1,j],[i+1,j],[i,j-1],[i,j+1]]
                for i,j in list:
                    if 0<=i<len(board) and 0<=j<len(board[0]) and (i,j) not in visited:
                        if dfs(word[1:],i,j, visited|{(i, j)}):
                            return True
        res = []
        for word in words:
            for i in range(len(board)):
                for j in range(len(board[0])):
                    if dfs(word,i,j,{(i, j)}):
                        res.append(word)
        return list(set(res))

方法二：前缀树

class Solution:
    def findWords(self, board: List[List[str]], words: List[str]) -> List[str]:
        trie = {}  # 构造字典树
        for word in words:
            node = trie
            for char in word:
                node = node.setdefault(char, {})
            node['#'] = True

        def search(i, j, node, pre, visited):  # (i,j)当前坐标，node当前trie树结点，pre前面的字符串，visited已访问坐标
            if '#' in node:  # 已有字典树结束
                res.add(pre)  # 添加答案
            for (di, dj) in ((-1, 0), (1, 0), (0, -1), (0, 1)):
                _i, _j = i+di, j+dj
                if -1 < _i < h and -1 < _j < w and board[_i][_j] in node and (_i, _j) not in visited:  # 可继续搜索
                    search(_i, _j, node[board[_i][_j]], pre+board[_i][_j], visited | {(_i, _j)})  # dfs搜索

        res, h, w = set(), len(board), len(board[0])
        for i in range(h):
            for j in range(w):
                if board[i][j] in trie:  # 可继续搜索
                    search(i, j, trie[board[i][j]], board[i][j], {(i, j)})  # dfs搜索
        return list(res)