【leetcode 字典树（前缀树） C++】1032. Stream of Characters

最新推荐文章于 2023-03-24 11:18:51 发布

原创最新推荐文章于 2023-03-24 11:18:51 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

288 篇文章

订阅专栏

本文介绍了一种使用Trie树的数据结构来实现字符流查询的方法。通过构建Trie树并逐字符查询的方式，可以高效地判断字符流中是否包含预先定义好的单词列表中的任意单词。文章详细解释了Trie树节点的定义、插入单词到Trie树的过程以及如何进行字符流的查询。

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

1032. Stream of Characters

在这里插入图片描述

class TrieNode {
public:
    bool is_end;
    vector<TrieNode*> children = vector<TrieNode*>(26, nullptr);
    TrieNode() { is_end = false; }
};


class StreamChecker {
public:
    TrieNode* root = new TrieNode();
    TrieNode* p;
    queue<TrieNode*> Q;

    void insert(string word) {
        p = root;
        for(int ii = 0; ii < word.size(); ii++) {
            if(!p->children[word[ii] - 'a']) p->children[word[ii] - 'a'] = new TrieNode();
            p = p->children[word[ii] - 'a'];
        }
        p->is_end = true;
    }

    StreamChecker(vector<string>& words) {
        for(auto &word : words) insert(word);
    }
    
    bool query(char letter) {
        bool flag = false;
        int Q_len = Q.size();
        for(int ii = 0; ii < Q_len; ii++) {
            p = Q.front();
            Q.pop();
            if(p->children[letter - 'a']) {
                Q.push(p->children[letter - 'a']);
                if(p->children[letter - 'a']->is_end) flag = true;
            }
        }
        if(root->children[letter - 'a']) {
            Q.push(root->children[letter - 'a']);
            if(root->children[letter - 'a']->is_end) flag = true;
        }
        return flag;
    }
};

/**
 * Your StreamChecker object will be instantiated and called as such:
 * StreamChecker* obj = new StreamChecker(words);
 * bool param_1 = obj->query(letter);
 */