212. 单词搜索 II

212. 单词搜索 II

原始题目链接：https://leetcode.cn/problems/word-search-ii/

给定一个 m x n 二维字符网格 board 和一个单词（字符串）列表 words， 返回所有二维网格上的单词 。

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

解题思路：

把单词列表构建成前缀树，然后二维数组在前缀树中查找单词。

代码实现：

class Solution:
    def findWords(self, board: List[List[str]], words: List[str]) -> List[str]:
        # 将words中的单词，构建成前缀树
        tree = {}
        for word in words:
            node = tree
            for c in word:
                # setdefault如果字典中不存在key
                # 会在字典中插入key并返回默认的value
                # 这里会返回默认的{}，赋值给node是c对应的{}
                # 所以会嵌套插入
                node = node.setdefault(c, {})
            # 赋值一个单词结束的标志位
            node['end'] = 1

        ans = []

        # 定义一个递归函数：dfs深度优先搜索
        # 用于board中的二维矩阵在前缀树中查找单词
        def dfs(row, col, tree, s):
            w = board[row][col]
            # 在前缀树中找到有该字符的那个子树
            tree = tree[w]
            if 'end' in tree and tree['end'] == 1:
                ans.append(s + w)
                # 防止重复加入
                tree['end'] = 0
            # 遍历过的字符暂时赋值为#，防止重复使用
            board[row][col] = '#'
            # 题意是相邻单元格：即再看当前字符的上下左右的字符
            for x, y in ([row - 1, col], [row + 1, col], [row, col - 1], [row, col + 1]):
                if 0 <= x < len(board) and 0 <= y < len(board[0]) and board[x][y] in tree:
                    dfs(x, y, tree, s + w)
            # 复原当前的单词
            board[row][col] = w 

        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] in tree:
                    dfs(i, j, tree, "")
        return ans

参考文献：
https://leetcode.cn/problems/word-search-ii/solution/qian-zhui-shu-dfs-by-powcai/