trie

最新推荐文章于 2024-10-27 21:16:57 发布

原创最新推荐文章于 2024-10-27 21:16:57 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

88 篇文章

订阅专栏

本文介绍了一种高效的数据结构——前缀树（Trie），详细解释了其存储原理、关键性质及实现方法。通过实例展示了如何使用前缀树进行单词插入、搜索及前缀匹配等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

trie，又称前缀树或字典树。它利用字符串的公共前缀来节约存储空间。
性质
（1）根节点不包含字符，除根节点外的每个节点只包含一个字符。
（2）从根节点到某一个节点，路径上经过的字符连接起来，为该节点对应的字符串。
（3）每个节点的所有子节点包含的字符串不相同。

class Trie(object):

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.root = {}
        self.word_end = -1

    def insert(self, word):
        """
        Inserts a word into the trie.
        :type word: str
        :rtype: None
        """
        curNode = self.root
        for c in word:            
            curNode = curNode.setdefault(c,{})
        curNode[self.word_end] = True

    def search(self, word):
        """
        Returns if the word is in the trie.
        :type word: str
        :rtype: bool
        """
        curNode = self.root
        for c in word:
            if not c in curNode:
                return False
            curNode = curNode[c]        
        return self.word_end in curNode

    def startsWith(self, prefix):
        """
        Returns if there is any word in the trie that starts with the given prefix.
        :type prefix: str
        :rtype: bool
        """
        curNode = self.root
        for c in prefix:
            if not c in curNode:
                return False
            curNode=curNode[c]
        return True


# Your Trie object will be instantiated and called as such:
# obj = Trie()
# obj.insert(word)
# param_2 = obj.search(word)
# param_3 = obj.startsWith(prefix)