【快乐水题】677. 键值映射

·  阅读 106
【快乐水题】677. 键值映射

原题:

力扣链接: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);
 */

复制代码

力扣结果展示:

在这里插入图片描述

分类:
后端
标签:
收藏成功!
已添加到「」, 点击更改