LeetCode 677 键值映射[Map] HERODING的LeetCode之路

Posted 2021-12-12 HERODING23

解题思路：
这道题可以开挂一波，反向套娃，你让我实现键值映射，那我就用键值映射实现，直接定义一个map，用来记录string和int对，sum函数实现时，通过substr来统计拥有prefix的值的和，代码如下：

class MapSum {
private:
    unordered_map<string, int> mp;
public:
    MapSum() {

    }
    
    void insert(string key, int val) {
        mp[key] = val;
    }
    
    int sum(string prefix) {
        int count = 0;
        for(auto& [k, v] : mp) {
            if(k.substr(0, prefix.size()) == prefix) {
                count += v;
            }
        }
        return count;
    }
};

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