Leetcode 720. 词典中最长的单词 (字典树搜索查找优化）

wwxy261

于 2022-01-06 00:07:45 发布

阅读量227

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/wwxy1995/article/details/122334886

版权

修改Trie中的搜索判断函数，要求每一个前缀都出现即可

class Trie {
private:
    vector<Trie*> children;
    bool isEnd;

public:
    Trie():children(26), isEnd(false) {}

    void insert(string& word) {
        Trie* node = this;
        for (auto c : word) {
            if (node->children[c - 'a'] == nullptr) {
                node->children[c - 'a'] = new Trie();
            }
            node = node->children[c - 'a'];
        }
        node->isEnd = true;
    }

    bool search(string& word) {
        Trie* node = this;
        for (auto c : word) {
            if (node->children[c - 'a'] == nullptr || !node->children[c-'a']->isEnd) {
                return false;
            }
            node = node->children[c -