Leetcode 677. 键值映射 (Trie树改造)

最新推荐文章于 2024-09-14 18:52:56 发布

wwxy261

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量562

点赞数

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

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

版权

在原始Tire树的基础上改造，每个节点维护后面的所有后缀的value之和。用一个HashMap计数处理重复。

struct TrieNode {
    int val;
    TrieNode * next[26];
    TrieNode() {
        this->val = 0;
        for (int i = 0; i < 26; ++i) {
            this->next[i] = nullptr;
        }
    }
};
class MapSum {
private:
    TrieNode* root;
    unordered_map<string, int> cnt; 
public:
    MapSum() {
        root = new TrieNode();
    }
    
    void insert(string key, int val) {
        int mines = (cnt.count(key) ? cnt[key] : 0);
        cnt[key] = val;
        TrieNode* node = root;
        for (auto c : key) {
            if(node->next[c - 'a'] == nullptr) {
                node->next[c - 'a'] = new TrieNode();
            }
            node = node->next[c - 'a'