字典树(前缀树)

class Trieroot:
    def __init__(self):
        self.isEnd = False
        self.next = {}
        self.pre = ""
        self.count = 0

class Trie:

    def __init__(self):
        self.root = Trieroot()

    def insert(self, word: str) -> None:
        node = self.root
        for st in word:
            ind = ord(st)-ord('a')
            if ind not in node.next:
                node.next[ind] = Trieroot() 
            node = node.next[ind]
        node.isEnd = True
        node.count +=1
        node.pre = word


    def search(self, word: str) -> bool:
        node = self.root

        for st in word:
            ind = ord(st)-ord('a')
            if ind not in node.next:
                return False
            node = node.next[ind]
        return node.isEnd
    
            
        
    
    def startsWith(self, prefix: str) -> bool:
        def find(node):
            if node.isEnd:
                save[node.pre] = node.count
                if not node.next:
                    return 
            for ind in node.next:
                find(node.next[ind])

        node = self.root
        for st in prefix:
            ind = ord(st)-ord('a')
            if ind not in node.next:
                return False
            node = node.next[ind]
        save = {}
        find(node)
        print(save)
        print(sorted(save,reverse=True))
            
        

# Your Trie object will be instantiated and called as such:
obj = Trie()
obj.insert("apple")
obj.insert("apple")
obj.insert("apple")
obj.insert("apple")
obj.insert("app")
obj.insert("app")
obj.insert("app")
obj.insert("sopo")
obj.insert("apm")
obj.search("app")

obj.startsWith("a")



#华为面试#
全部评论
了解字典树
点赞 回复 分享
发布于 2022-10-20 14:41 河南

相关推荐

nbdy:字太多了,写简历不是写自传,亮点难点技能点列出来就行,要简明扼要
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务