题目描述
实现一个 MapSum 类,支持两个方法,insert 和 sum:
- MapSum():初始化 MapSum 对象
- void insert(String key, int val): 插入 key-val 键值对,字符串表示键 key ,整数表示值 val 。如果键 key 已经存在,那么原来的键值对将被替代成新的键值对。
- int sum(string prefix) :返回所有以该前缀 prefix 开头的键 key 的值的总和。
示例:
输入:
[“MapSum”, “insert”, “sum”, “insert”, “sum”] [[], [“apple”, 3],
[“ap”], [“app”, 2], [“ap”]] 输出: [null, null, 3, null, 5]解释:
MapSum mapSum = new MapSum();
mapSum.insert(“apple”, 3);
mapSum.sum(“ap”); // return 3 (apple = 3)
mapSum.insert(“app”, 2);
mapSum.sum(“ap”); // return 5 (apple + app = 3 + 2 = 5)
提示:
- 1 <= key.length, prefix.length <= 50
- key 和 prefix 仅由小写英文字母组成
- 1 <= val <= 1000
- 最多调用 50 次 insert 和 sum
提示:以下是本篇文章正文内容,下面案例可供参考
一、解题思路
我们可以直接暴力扫描,存储所有键值对,根据每次给定的前缀,依次搜索所有的键值。如果键值包含给定的前缀,就将对应的val 相加,返回所有符合要求的 val 的和。
二、完整代码
class MapSum {
Map<String, Integer> map;
public MapSum() {
map = new HashMap<>();
}
public void insert(String key, int val) {
map.put(key,val);
}
public int sum(String prefix) {
int res = 0;
for (String s : map.keySet()) {
if (s.startsWith(prefix)) {
res += map.get(s);
}
}
return res;
}
}