208-Implement Trie(Prefix Tree)
题目:实现一个字典树
class Trie(object): def __init__(self): self.root = {} self.end_of_word = '#' def insert(self,word): node = self.root for char in word: node = node.setdefault(char,{}) node[self.end_of_word] = self.end_of_word def search(self,word): node = self.root for char in word: if char not in node: return False node = node[char] return self.end_of_word in node def startWith(self,prefix): node = self.root for char in prefix: if char not in node: return False node = node[char] return True
注:
使用字典结构来存储,每层都是一个字典,存储此层所有的元素(包括字典)。每个元素存储方式也为一个字典,key为字符和一个标记end_of_word,表示该字符是否为一个单词的结束。插入操作时,遍历要插入词的每个字符,如果在字典树中找不到,则将该字符插入字典树,默认值为{},如果查找到,则继续往下查找,遍历结束,将最后一个字符的标志位设置。查找操作时,遍历单词的每个字符,如果遇到每个字符差找不到,说明不在该字典树,如果查找到,在判断该字符的标记符是否有。