208_实现Trie(前缀树)

"""
实现一个 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 构成的。
保证所有输入均为非空字符串。
"""


class Trie:

    def __init__(self):
        self.root = {}

    def insert(self, word):
        node = self.root
        for s in word:
            if s in node.keys():
                node = node[s]
            else:
                node[s] = {}
                node = node[s]
        node['is_word'] = True

    def search(self, word):
        node = self.root
        for s in word:
            if s in node.keys():
                node = node[s]
            else:
                return False
        if 'is_word' in node.keys():
            return True
        else:
            return False

    def startsWith(self, prefix):
        node = self.root
        for s in prefix:
            if s in node.keys():
                node = node[s]
            else:
                return False
        return True