LeetCode 208 实现 Trie (前缀树)

hewesH

于 2021-12-29 15:52:42 发布

阅读量89

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/cxx05260/article/details/122217816

leetcode 专栏收录该内容

64 篇文章

订阅专栏

这个博客介绍了字典树（Trie）的数据结构及其应用。通过`searchPrefix`方法实现前缀搜索，`insert`方法用于插入单词构建字典树，而`search`和`startsWith`方法分别检查字典树中是否存在完整单词和是否以指定前缀开始。字典树在字符串查找和数据压缩等领域有广泛应用。

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

class Trie:
    def __init__(self):
        self.children = [None] * 26
        self.isEnd = False

    def searchPrefix(self, prefix):
        node = self
        for ch in prefix:
            ch = ord(ch) - ord("a")
            if not node.children[ch]:
                return None
            node = node.children[ch]
        return node

    # 根据插入的word构造字典树
    def insert(self, word):
        node = self
        for ch in word:
            ch = ord(ch) - ord("a")
            if not node.children[ch]:
                node.children[ch] = Trie()
            node = node.children[ch]
        node.isEnd = True

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

    def startsWith(self, prefix):
        return self.searchPrefix(prefix) is not None