【刷爆LeetCode】七月算法集训（23）字典树

最新推荐文章于 2024-08-10 22:52:22 发布

...1999

最新推荐文章于 2024-08-10 22:52:22 发布

阅读量119

点赞数

分类专栏：七月算法集训文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/LAdestiny/article/details/125949505

版权

七月算法集训专栏收录该内容

31 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用字典树（Trie）解决LeetCode中的1032.字符流问题。作者首先给出了题目描述，然后详细阐述了解题思路，即对输入字符串逆序后插入字典树，并在字符流中查找后缀200个字符的匹配。接着，展示了一段C++代码实现，包括TrieNode和Trie类的定义以及插入和搜索功能。最后，作者鼓励读者每日练习算法，以期在一个月内取得进步。

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

题目来源于知识星球—英雄算法联盟，七月算法集训专题

前言

跟随英雄算法联盟博主—英雄哪里出来，每天完成相应的算法练习，一个月后，必定会有所成长！
一道题足矣学习字典树（模板）！

一、1032.字符流（困难）

1.题目描述

2.解题思路

对给出的字符串进行逆序后插入树中，取后缀200个字符，搜索即可。

3.代码演示（C++）

class StreamChecker 
{
    class TrieNode 
    {
    public:
        int end;
        TrieNode *next[26];
        TrieNode() 
        {
            end = 0;
            memset(next, 0, sizeof(next));
        }
    };
    class Trie 
    {
    public:
        TrieNode* root;
        Trie() 
        {
            root = new TrieNode();
        }
        void insert(string word) 
        {
            TrieNode *now = root;
            for(int i = 0; i < word.size(); ++i) 
            {
                int child = word[i] - 'a';
                if( nullptr == now->next[child] ) 
                {
                    now->next[child] = new TrieNode();
                }
                now = now->next[child];
            }
            ++now->end;
        }
        bool search(string word) 
        {
            TrieNode *now = root;
            for(int i = 0; i < word.size(); ++i) 
            {
                int child = word[i] - 'a';
                if(now->end) 
                {
                    return true;
                }
                if( nullptr == now->next[child]) 
                {
                    return false;
                }
                now = now->next[child];
            }
            return now->end;
        }
    };
public:
    Trie T;
    string s;
    StreamChecker(vector<string>& words) 
    {
        for(int i = 0; i < words.size(); ++i) 
        {
            reverse(words[i].begin(), words[i].end());
            T.insert(words[i]);
        }
    }
    bool query(char letter) 
    {
        s += letter;
        string findS = "";
        for(int i = s.size()-1; i >= 0; --i) 
        {
            findS += s[i];
            if(findS.size() > 200) break;
        }
        return T.search(findS);
    }
};