LeetCode 677. 键值映射

一、题目

1、题目描述

  设计一个 map，满足以下几点:
  字符串表示键，整数表示值，返回具有前缀等于给定字符串的键的值的总和
  实现一个 MapSum类：
    MapSum()初始化 MapSum对象
    void insert(String key, int val)插入 key-val键值对，字符串表示键 key，整数表示值 val。如果键 key已经存在，那么原来的键值对 key-value将被替代成新的键值对。
    int sum(string prefix)返回所有以该前缀 prefix开头的键 key的值的总和。
  样例输入： ["MapSum", "insert", "sum", "insert", "sum"] [[], ["apple", 3], ["ap"], ["app", 2], ["ap"]]
  样例输出： [null, null, 3, null, 5]

2、基础框架

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

3、原题链接

LeetCode 677. 键值映射
剑指 Offer II 066. 单词之和

二、解题报告

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);
 */

---

三、本题小知识

  看到前缀，首先想到字典树。

---

四、加群须知

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

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

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

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜