677. 键值映射
实现一个 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
解题思路
使用字典树维护插入的键值对
-
新建Node类,用来构建字典树,每个节点拥有26个子节点,对应26个字母,若某个节点为空,则代表不存在包含该字母的前缀。在每个节点中记录,满足当前节点前缀的总和的大小。
-
对于前缀 prefix 开头的键 key 的值的总和的处理,每次在字典树上插入节点的时候,我们都在插入路径的每一个节点上增加val,而替换key值的时候,我们先正常插入key值,当发现key值已经被插入过了,就沿着插入路径给每个节点减去val。
代码
class Node {
public:
Node* child[26] ;
int sum,val;
bool is_end;
};
class MapSum {
public:
Node *root;
MapSum() {
root=new Node();
}
void insert(string key, int val) {
Node *cur = root;
for (int i = 0; i < key.size(); ++i) {
if (cur->child[key[i] - 'a'] == nullptr)
cur->child[key[i] - 'a'] = new Node();
cur = cur->child[key[i] - 'a'];
if (cur != nullptr)
(cur->sum) += val;
}
if (cur->is_end) {
Node *new_cur = root;
for (int i = 0; i < key.size(); ++i) {
new_cur = new_cur->child[key[i] - 'a'];
if (new_cur != nullptr)
(new_cur->sum) -= cur->val;
}
}
cur->is_end=true;
cur->val = val;
}
int sum(string prefix) {
Node *cur=root;
int sum(0);
int i = 0;
for (; i < prefix.size()&&cur!= nullptr; ++i) {
cur=cur->child[prefix[i]-'a'];
if (cur!= nullptr)
sum=cur->sum;
}
return cur!= nullptr?sum:0;
}
};
/**
* Your MapSum object will be instantiated and called as such:
* MapSum* obj = new MapSum();
* obj->insert(key,val);
* int param_2 = obj->sum(prefix);
*/