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

posted @ 2019-08-19 22:01  尘世中一个迷途小书童  阅读(131)  评论(0编辑  收藏  举报