快乐水题677. 键值映射

Posted 2021-12-13 谁吃薄荷糖

原题：

力扣链接:677. 键值映射

题目简述：

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

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

解题思路

1.利用map来做map;
2.查询前缀要使用substr()，第一把没仔细审题使用了find()提交失败了；
3.over;

C++代码：

class MapSum {
public:
    MapSum() {
       
    }
    
    void insert(string key, int val) {
        umap[key] = val;
    }
    
    int sum(string prefix) {
        int ret = 0;

        int i = 0;
        for(auto it = umap.begin(); it != umap.end(); it++)
        {
            if(it->first.substr(0, prefix.size()) == prefix)
            {
                ret += it->second;
            }
        }
        return ret;
    }

private:
     unordered_map<string, int> umap;
};

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

力扣结果展示：