原文链接: https://leetcode-cn.com/problems/z1R5dt

中文题目

实现一个 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

 

注意：本题与主站 677 题相同： https://leetcode-cn.com/problems/map-sum-pairs/

通过代码

高赞题解

前缀树

此题还是使用前缀树的解法，只是将常规的前缀树节点中，记录是否为插入字符串结尾的 isword 改成记录字符串值的 val。前缀树还要实现类函数 insert 实现插入字符串，以及类函数 coutSum 实现返回所有以该前缀开头的字符串的值的总和。在实现 coutSum 函数时，可以采用先遍历到该前缀的尾节点，若不存在该前缀的路径则返回0，若存在则从尾节点开始使用广度优先搜索算法，实现返回所有以该前缀开头的字符串的值的总和。完整代码如下：

// 构造前缀树节点
class Trie {
public:
    int val;
    vector<Trie*> children;
    Trie () : val(0), children(26, nullptr) {}

    // 实现插入字符串
    void insert(string& str, int m) {
        Trie* node = this;
        for (auto& ch : str) {
            if (node->children[ch - 'a'] == nullptr) {
                node->children[ch - 'a'] = new Trie();
            }
            node = node->children[ch - 'a'];
        }
        node->val = m;
    }

    // 实现返回所有以该前缀 prefix 开头的键 key 的值的总和
    int coutSum(string &prefix) {
        Trie* node = this;
        for (auto& ch : prefix) {
            if (node->children[ch - 'a'] == nullptr) {
                return 0;
            }
            node = node->children[ch - 'a'];
        }

        // BFS
        int count = 0;
        queue<Trie*> que;
        que.push(node);
        while (!que.empty()) {
            Trie* node = que.front();
            que.pop();
            count += node->val;
            for (int i = 0; i < node->children.size(); ++i) {
                if (node->children[i] != nullptr) {
                    que.push(node->children[i]);
                }
            }
        }
        return count;
    }
};

class MapSum {
private:
    Trie* root;
public:
    MapSum() {
        root = new Trie();
    }
    
    void insert(string key, int val) {
        root->insert(key, val);
    }
    
    int sum(string prefix) {
        return root->coutSum(prefix);
    }
};

统计信息

通过次数	提交次数	AC比率
1966	3023	65.0%

提交历史

提交时间	提交结果	执行时间	内存消耗	语言