LeetCode 力扣每日一题题解 677. 键值映射

最新推荐文章于 2024-11-10 14:06:08 发布

axianera

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量3.8k

点赞数

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

本文链接：https://blog.csdn.net/m0_56937307/article/details/121315045

版权

LeetCode 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用C++实现一个MapSum类，该类支持insert方法插入键值对并更新，以及sum方法计算所有以特定前缀开头的键的值之和。通过一个unordered_map存储键值对，利用字符串的substr方法检查前缀匹配，从而高效地完成求和操作。

摘要由CSDN通过智能技术生成

原题描述：

实现一个 MapSum 类，支持两个方法，insert 和 sum：

MapSum() 初始化 MapSum 对象
void insert(String key, int val) 插入 key-val 键值对，字符串表示键 key ，整数表示值 val 。如果键 key 已经存在，那么原来的键值对将被替代成新的键值对。
int sum(string prefix) 返回所有以该前缀 prefix 开头的键 key 的值的总和。

原题链接：677. 键值映射 - 力扣（LeetCode） (leetcode-cn.com)

C++代码：

class MapSum {
public:
    MapSum() {

    }
    
    void insert(string key, int val) {
        cnt[key] = val;
    }
    
    int sum(string prefix) {
        int res = 0;
        for (auto & [key,val] : cnt) {
            if (key.substr(0, prefix.size()) == prefix) {
                res += val;
            }
        }
        return res;
    }
private:
    unordered_map<string, int> cnt;
};