python 实现 trie树（字典树）

最新推荐文章于 2023-12-04 08:43:13 发布

arv1nChen

最新推荐文章于 2023-12-04 08:43:13 发布

阅读量555

点赞数

分类专栏： NLP python 文章标签： python

本文链接：https://blog.csdn.net/u014403221/article/details/125104503

版权

NLP 同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

python

16 篇文章 0 订阅

订阅专栏

本文介绍了Trie树（又称字典树或前缀树），它是一种用于存储字符串的数据结构。文章通过类`Trie`展示了如何实现Trie树的插入、搜索和检查字符串是否以特定前缀开始的功能。Trie树的关键特性是节点的所有后代具有共同的前缀。

摘要由CSDN通过智能技术生成

tire 树也叫字典树，也是一种 N 叉树，是一种特殊的前缀树结构。

1、前缀树是N叉树的一种特殊形式。通常来说，一个前缀树是用来存储字符串的。前缀树的每一个节点代表一个字符串（前缀）。每一个节点会有多个子节点，通往不同子节点的路径上有着不同的字符。子节点代表的字符串是由节点本身的原始字符串，以及通往该子节点路径上所有的字符组成的。

下面是前缀树的一个例子：

前缀树的一个重要的特性是，节点所有的后代都与该节点相关的字符串有着共同的前缀。这就是前缀树名称的由来。

2、实现tire 树的插入和搜索、以及根据前缀判断是否在trie树操作方式：

class Trie:
 
    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.root = {}
        self.end = -1
 
    def insert(self, word):
        """
        Inserts a word into the trie.
        :type word: str
        :rtype: void
        """
        
        curNode = self.root
        for c in word:
            if not c in curNode:
                curNode[c] = {}
            curNode = curNode[c]
        curNode[self.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]
            
        # Doesn't end here
        if not self.end in curNode:
            return False
        
        return True
 
    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)