211 添加与搜索单词 - 数据结构设计（Trie树 + dfs）

1. 问题描述：

请你设计一个数据结构，支持添加新单词和查找字符串是否与任何先前添加的字符串匹配 。实现词典类 WordDictionary ：WordDictionary() ；初始化词典对void addWord(word) 将 word 添加到数据结构中，之后可以对它进行匹配；bool search(word) 如果数据结构中存在字符串与 word 匹配，则返回 true ；否则，返回  false 。word 中可能包含一些 '.' ，每个 . 都可以表示任何一个字母。

示例：

输入：
["WordDictionary","addWord","addWord","addWord","search","search","search","search"]
[[],["bad"],["dad"],["mad"],["pad"],["bad"],[".ad"],["b.."]]
输出：
[null,null,null,null,false,true,true,true]
解释：
WordDictionary wordDictionary = new WordDictionary();
wordDictionary.addWord("bad");
wordDictionary.addWord("dad");
wordDictionary.addWord("mad");
wordDictionary.search("pad"); // return False
wordDictionary.search("bad"); // return True
wordDictionary.search(".ad"); // return True
wordDictionary.search("b.."); // return True

提示：

1 <= word.length <= 500
addWord 中的 word 由小写英文字母组成
search 中的 word 由 '.' 或小写英文字母组成
最多调用 50000 次 addWord 和 search

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/design-add-and-search-words-data-structure

2. 思路分析：

这道题目与Trie树的题目是类似的，唯一不同的是这里搜索单词的时候是不一样的，这里搜索单词的时候可能会存在.字符，.字符可以表示任意一个字符，我们可以先构建成Trie树，初始化和添加单词到Trie树的操作都是一样的，对于搜索单词的操作我们特殊处理一下，因为.可以匹配任意一个字符，所以当我们遇到.的时候搜索当前这一层所有可能的字母，也即存在多个可能搜索的分支，我们可以使用一个递归方法来搜索所有可能的分支即可（递归尝试所有可能的分支），当遇到字母不匹配的时候那么直接返回False即可。

3. 代码如下：

class WordDictionary:
    def __init__(self):
        # 初始化当前dic属性
        self.dic = {}

    def addWord(self, word: str) -> None:
        t = self.dic
        for c in word:
            if c not in t:
                t[c] = {}
            t = t[c]
        # 单词结尾的标记
        t["end"] = 1

    def dfs(self, word: str, dic: dict, i: int):
        if i == len(word): return "end" in dic
        # 当前字符不是.
        if word[i] != ".":
            if word[i] not in dic:
                return False
            return self.dfs(word, dic[word[i]], i + 1)
        else:
            for key in dic.keys():
                # 这里使用字典的一个坑是需要判断一下当前字典的键是否是单词结尾的标记没有加上这个条件在搜索的时候会出现错误
                if key != "end":
                    if self.dfs(word, dic[key], i + 1): return True
            return False

    # 因为当前有.所以需要使用递归来搜索每一个可能的分支
    def search(self, word: str) -> bool:
        return self.dfs(word, self.dic, 0)


# if __name__ == '__main__':
    # wordDictionary = WordDictionary()
    # wordDictionary.addWord("a")
    # wordDictionary.addWord("a")
    # print(wordDictionary.search("."))
    # print(wordDictionary.search("aa"))
    # print(wordDictionary.search("a"))
    # print(wordDictionary.search(".a"))
    # print(wordDictionary.search("a."))