[leetcode]208. Implement Trie (Prefix Tree)

最新推荐文章于 2024-01-22 12:36:44 发布

原创最新推荐文章于 2024-01-22 12:36:44 发布 · 247 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 3 个专栏收录

473 篇文章

订阅专栏

树

60 篇文章

订阅专栏

字符串处理

38 篇文章

订阅专栏

本文介绍了一个Trie（字典树或前缀树）的数据结构实现，包括插入、搜索和查找前缀的方法。该实现假设所有输入由小写字母组成，并通过节点类和Trie类来构建树结构。

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

题目链接：https://leetcode.com/problems/implement-trie-prefix-tree/tabs/description

Implement a trie with insert, search, and startsWith methods.

Note:
You may assume that all inputs are consist of lowercase letters a-z.

class TrieNode{
public:
    TrieNode* next[26];
    bool is_word;

    TrieNode(bool b=false)
    {
        memset(next,0,sizeof(next));
        is_word=b;
    }
};

class Trie {
public:
    /** Initialize your data structure here. */
    Trie() {
        root=new TrieNode();
    }

    /** Inserts a word into the trie. */
    void insert(string word) {
        TrieNode* p=root;
        for(int i=0;i<word.size();i++)
        {
            if(p->next[word[i]-'a']==NULL)
                p->next[word[i]-'a']=new TrieNode();
            p=p->next[word[i]-'a'];
        }
        p->is_word=true;
    }

    /** Returns if the word is in the trie. */
    bool search(string word) {
        TrieNode* p=find(word);
        return p!=NULL && p->is_word;
    }

    /** Returns if there is any word in the trie that starts with the given prefix. */
    bool startsWith(string prefix) {
        return find(prefix)!=NULL;
    }
private:
    TrieNode* root;
    TrieNode* find(string key)
    {
        TrieNode* p=root;
        for(int i=0;i<key.size() && p!=NULL;i++)
            p=p->next[key[i]-'a'];
        return p;
    }
};