剑指offer_066 单词之和

题目：

实现一个 MapSum 类，支持两个方法，insert 和 sum：

MapSum() 初始化 MapSum 对象
void insert(String key, int val) 插入 key-val 键值对，字符串表示键 key ，整数表示值 val 。如果键 key 已经存在，那么原来的键值对将被替代成新的键值对。
int sum(string prefix) 返回所有以该前缀 prefix 开头的键 key 的值的总和。

示例：

输入：
inputs = ["MapSum", "insert", "sum", "insert", "sum"]
inputs = [[], ["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

代码：

class MapSum {

    class TreeNode {
        TreeNode[] next = new TreeNode[26];
        int sum = 0;

        public void add(String str, int val) {
        TreeNode cur = root;
        for (char c : str.toCharArray()) {
            int index = c - 'a';
            if (cur.next[index] == null) {
                cur.next[index] = new TreeNode();
            }
            cur.next[index].sum += val;
            cur = cur.next[index];
        }
    }

        public int search(String str) {
            TreeNode cur = root;
            for (char c : str.toCharArray()) {
                int index = c - 'a';
                if (cur.next[index] == null) {
                    return 0;
                }
                cur = cur.next[index];
            }
            return cur.sum;
        }
    }
    
    TreeNode root;
    Map<String, Integer> map;
    /** Initialize your data structure here. */
    public MapSum() {
        root = new TreeNode();
        map = new HashMap<>();
    }

    
    
    public void insert(String key, int val) {
        root.add(key, val - map.getOrDefault(key, 0));
        map.put(key, val);
    }


    
    public int sum(String prefix) {
        return root.search(prefix);
    }
}

/**
 * Your MapSum object will be instantiated and called as such:
 * MapSum obj = new MapSum();
 * obj.insert(key,val);
 * int param_2 = obj.sum(prefix);
 */

解题思路：

依旧是前缀树，只不过在添加的时候加上权值就可以啦，搜索的时候返回权值就可以啦。注意在如果重复添加两个一样的字符串时要 root.add(key, val - map.getOrDefault(key, 0));

注意：

记得按时吃饭。

参考链接：

力扣