677. 键值映射

这是我参与11月更文挑战的第13天，活动详情查看：2021最后一次更文挑战

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

解题思路

使用字典树维护插入的键值对

1. 新建Node类，用来构建字典树，每个节点拥有26个子节点，对应26个字母，若某个节点为空，则代表不存在包含该字母的前缀。在每个节点中记录，满足当前节点前缀的总和的大小。

2. 对于前缀 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);
 */