文章目录

• 一、题目
• • 1、题目描述
  • 2、基础框架
  • 3、原题链接
• 二、解题报告
• • 1、思路分析
  • 2、时间复杂度
  • 3、代码详解
• 三、本题小知识
• 四、加群须知

一、题目

1、题目描述

2、基础框架

• C++ 版本给出的基础框架代码如下：

3、原题链接

二、解题报告

1、思路分析

  $(1)$ 利用字典树执行插入，并且在尾部结点更新键对应的值。
  $(2)$ 查找，就是顺着字符串一直找，直到找到字符串结束，并且对应给定结点的子树统计它所有的值的和。

2、时间复杂度

   最坏时间复杂度 $O(n)$ 。

3、代码详解

class TrieNode {
public:
    int sum;
    TrieNode *next[26];   // 指向所有的子结点

    TrieNode() {
        sum = 0;
        memset(next, 0, sizeof(next));
    }
};

class Trie {
    TrieNode* root;
public:
    Trie() {
        root = new TrieNode();
    }
    
    void insert(string word, int val) {
        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->sum = val;
    }

    int dfs(TrieNode* root) {
        if(root == nullptr) {
            return 0;
        }
        int sum = root->sum;
        for(int i = 0; i < 26; ++i) {
            sum += dfs(root->next[i]);
        }
        return sum;
    }
    
    int searchPrefixSum(string word) {
        TrieNode *now = root;
        for(int i = 0; i < word.size(); ++i) {
            int child = word[i] - 'a';
            if( nullptr == now->next[child]) {
                return 0;
            }
            now = now->next[child];
        }
        return dfs(now);
    }

};


class MapSum {
    Trie tree;
public:
    /** Initialize your data structure here. */
    MapSum() {
        
    }
    
    void insert(string key, int val) {
        tree.insert(key, val);
    }
    
    int sum(string prefix) {
        return tree.searchPrefixSum(prefix);
    }
};

/**
 * Your MapSum object will be instantiated and called as such:
 * MapSum* obj = new MapSum();
 * obj->insert(key,val);
 * int param_2 = obj->sum(prefix);
 */

三、本题小知识

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥