题目:实现一个字典树
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,表示该字符是否为一个单词的结束。插入操作时,遍历要插入词的每个字符,如果在字典树中找不到,则将该字符插入字典树,默认值为{},如果查找到,则继续往下查找,遍历结束,将最后一个字符的标志位设置。查找操作时,遍历单词的每个字符,如果遇到每个字符差找不到,说明不在该字典树,如果查找到,在判断该字符的标记符是否有。