[数据结构]字典树

最新推荐文章于 2024-09-16 18:31:59 发布

无良小生

最新推荐文章于 2024-09-16 18:31:59 发布

阅读量367

点赞数

文章标签：数据结构

本文链接：https://blog.csdn.net/WuLiangXiaoSeng/article/details/124546470

版权

LeetCode 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文介绍了字典树（Trie）的基本概念，如何通过树状结构存储字符串，以及如何利用它进行快速前缀匹配、搜索和判断字符串是否以某个前缀开头。通过实例和Python代码实现展示了Trie的数据结构和操作方法。

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

字典树

定义

简单来说，字典树就是将 大量的字符串 用树的形式存储起来，这种存储结构适用于查询多。具体实例

解析

实现

class Trie:

    def __init__(self):
        self.childern = [None for i in range(26)]
        self.isEnd = False
    
    def insert(self, word: str) -> None:
        node = self
        for ch in word:
            ch = ord(ch) - 97       # 97 为ord("a")
            if not node.childern[ch]:
                node.childern[ch] = Trie()
            node = node.childern[ch]
        node.isEnd = True

    # 定义一个查找前缀的函数，方便下面的实现
    def searchPrefix(self, prefix: str) -> "Trie":
        node = self
        for ch in prefix:
            ch = ord(ch) - 97       # 97 为ord("a")
            if not node.childern[ch]:
                return None
            node = node.childern[ch]
        return node

    def search(self, word: str) -> bool:
        node = self.searchPrefix(word)
        return node is not None and node.isEnd

    def startsWith(self, prefix: str) -> bool:
        node = self.searchPrefix(prefix)
        return node is not None
ix(prefix)
        return node is not None