码迷,mamicode.com
首页 > 其他好文 > 详细

208-Implement Trie(Prefix Tree)

时间:2019-08-19 23:14:29      阅读:86      评论:0      收藏:0      [点我收藏+]

标签:一个   highlight   key   init   包括   for   default   start   elf   

题目:实现一个字典树

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

208-Implement Trie(Prefix Tree)

标签:一个   highlight   key   init   包括   for   default   start   elf   

原文地址:https://www.cnblogs.com/kingshine007/p/11380027.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!