剑指 Offer II 062. 实现前缀树

花飞雨追

于 2021-12-31 11:17:00 发布

阅读量1.1k

点赞数

分类专栏： Leetcode 数据结构-字典树文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/qq_36258516/article/details/122252527

版权

Leetcode 同时被 2 个专栏收录

142 篇文章

订阅专栏

数据结构-字典树

6 篇文章

订阅专栏

本文详细介绍了一个基于Python实现的前缀树（Trie）模板，包括初始化、插入单词、搜索和判断前缀的功能。通过实例演示，理解Trie在字符串操作中的高效性和应用场景，如单词查找、自动补全和字符串匹配。

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

解题思路

前缀树模板。

代码

class Trie {
    private Trie[] next;
    boolean isEnd;

    /**
     * Initialize your data structure here.
     */
    public Trie() {
        next = new Trie[26];
        isEnd = false;
    }

    /**
     * Inserts a word into the trie.
     */
    public void insert(String word) {
        int length = word.length();
        Trie now = this;
        for (int i = 0; i < length; i++) {
            int index = word.charAt(i) - 'a';
            if (now.next[index] == null) {
                now.next[index] = new Trie();
            }
            now = now.next[index];
        }
        now.isEnd = true;
    }

    /**
     * Returns if the word is in the trie.
     */
    public boolean search(String word) {
        int length = word.length();
        Trie now = this;
        for (int i = 0; i < length; i++) {
            int index = word.charAt(i) - 'a';
            if (now.next[index] == null) return false;
            now = now.next[index];
        }
        return now.isEnd;
    }

    /**
     * Returns if there is any word in the trie that starts with the given prefix.
     */
    public boolean startsWith(String prefix) {
        int length = prefix.length();
        Trie now = this;
        for (int i = 0; i < length; i++) {
            int index = prefix.charAt(i) - 'a';
            if (now.next[index] == null) return false;
            now = now.next[index];
        }
        return true;
    }
}