【Trie】前缀树的三种操作（力扣面试经典150）

最新推荐文章于 2024-07-11 15:05:23 发布

敲击石头的一百零一次

最新推荐文章于 2024-07-11 15:05:23 发布

阅读量252

点赞数 5

分类专栏： ds 文章标签： leetcode 面试 c# c++

本文链接：https://blog.csdn.net/2202_75999061/article/details/140309757

版权

ds 专栏收录该内容

83 篇文章 0 订阅

订阅专栏

一、题目

. - 力扣（LeetCode）. - 备战技术面试？力扣提供海量技术面试资源，帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/implement-trie-prefix-tree/description/?envType=study-plan-v2&envId=top-interview-150

二、题解

class Trie {
public:
    int son[1000010][26], cnt[1000010], idx;
    Trie() {
        idx = 0;
        memset(son, 0, sizeof(son));
        memset(cnt, 0, sizeof(cnt));
    }

    ~Trie() {
        idx = 0;
        memset(son, 0, sizeof(son));
        memset(cnt, 0, sizeof(cnt));
    }
    
    void insert(string word) {
        int p = 0;
        for (auto x : word) {
            int child = x - 'a';
            if (!son[p][child]) son[p][child] = ++idx;
            p = son[p][child];
        }
        cnt[p]++;
    }
    
    bool search(string word) {
        int p = 0;
        for (auto x : word) {
            int child = x - 'a';
            if (!son[p][child]) return 0;
            p = son[p][child];
        }
        return cnt[p];
    }
    
    bool startsWith(string prefix) {
        int p = 0;
        for (auto x : prefix) {
            int child = x - 'a';
            if (!son[p][child]) return 0;
            p = son[p][child];
        }
        return 1;
    }
};

/**
 * Your Trie object will be instantiated and called as such:
 * Trie* obj = new Trie();
 * obj->insert(word);
 * bool param_2 = obj->search(word);
 * bool param_3 = obj->startsWith(prefix);
 */