【快乐水题】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);
*/
力扣结果展示:
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
二维码