基于python的dict实现Trie树O(1)时间复杂度

题目：

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert(“apple”);
trie.search(“apple”); // 返回 true
trie.search(“app”); // 返回 false
trie.startsWith(“app”); // 返回 true
trie.insert(“app”);
trie.search(“app”); // 返回 true
说明:

你可以假设所有的输入都是由小写字母 a-z 构成的。
保证所有输入均为非空字符串。

思路：

insert、research、startWith都是O(1)时间复杂度

代码：

class Trie:

    def __init__(self):
        self.children = {}
        self.EndOfWord = '#'

    def insert(self, word: str) -> None:
        # 这样定义：键为字母，值为字典，字典的键又为字母，值又是字典，直到次单词最后一个字母的值所对应的键是#，insert完毕，除了最后一个#，其他的从顶层遍历下来就是一个单词，所以也叫做键树
        helper = self.children
        for v in word:
            # helper[v] = {} 这样写再来一个前缀相同的会覆盖掉之前的
            helper[v] = helper.setdefault(v, {})
            helper = helper[v]
        # 最后的键'#' 随便赋一个值，不然报错
        helper[self.EndOfWord] = 0

    def search(self, word: str) -> bool:
        helper = self.children
        for v in word:
            if v in helper:
                helper = helper[v]
            else:
                return False
        # 处理给的实例中第四行 trie.search("app");// returns false
        return '#' in helper

    def startsWith(self, prefix: str) -> bool:
        helper = self.children
        for v in prefix:
            if v in helper:
                helper = helper[v]
            else:
                return False
        return True