Leetcode 208. 实现 Trie (前缀树) —— 重要

JehanRio

已于 2023-09-23 09:57:44 修改

阅读量121

点赞数

分类专栏： Leetcode专栏文章标签：算法 c++

于 2023-05-22 10:17:13 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/130801743

版权

Leetcode专栏专栏收录该内容

278 篇文章 0 订阅

订阅专栏

文章目录

题目
代码（首学前缀树）
代码（9.23 二刷看解析）

题目

在这里插入图片描述

Leetcode 208. 实现 Trie (前缀树)

代码（首学前缀树）

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

    Trie* searchPrefix(string prefix) {
        Trie* node = this;
        for(char c : prefix) {
            c -= 'a';
            if(node->children[c] == nullptr) {
                return nullptr;
            }
            node = node->children[c];
        }
        return node;
    }
public:

    Trie() : children(26), isEnd(false) {}
    
    void insert(string word) {
        Trie* node = this;
        for(char c : word) {
            c -= 'a';
            if(node->children[c] == nullptr) {
                node->children[c] = new Trie;
            }
            node = node->children[c];
        }
        node->isEnd = true;
    }
    
    bool search(string word) {
        Trie* node = searchPrefix(word);
        return node && node->isEnd;
    }
    
    bool startsWith(string prefix) {
        return searchPrefix(prefix);
    }
};

代码（9.23 二刷看解析）

进一步熟悉前缀树

class Trie {
private:
    vector<Trie*> children;
    bool isEnd;
    Trie* searchPrefix(string prefix) {
        Trie* node = this;
        for(char c : prefix) {
            c -= 'a';
            if(node->children[c] == nullptr) {
                return nullptr;
            }
            node = node->children[c];
        }
        return node;
    }
public: 
    Trie() : children(26), isEnd(false){}
    
    void insert(string word) {
        Trie* node = this;
        for(char c : word) {
            c -= 'a';
            if(node->children[c] == nullptr) {
                node->children[c] = new Trie;
            }
            node = node->children[c];
        }
        node->isEnd = true;
    }
    
    bool search(string word) {
        auto node = searchPrefix(word);
        return node && node->isEnd;
    }
    
    bool startsWith(string prefix) {
        return searchPrefix(prefix);
    }
};