LeetCode - Map Sum Pairs

最新推荐文章于 2021-07-24 04:15:01 发布

real_lisa

最新推荐文章于 2021-07-24 04:15:01 发布

阅读量118

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/real_lisa/article/details/89081994

LeetCode 专栏收录该内容

180 篇文章

订阅专栏

本文介绍了两种实现MapSum数据结构的方法。第一种使用unordered_map来高效处理字符串键的插入和求和操作，第二种则利用map进行键值对的管理及前缀匹配计数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解法一：unordered_map<string, pair<int,int>>

class MapSum {
public:
    /** Initialize your data structure here. */
    MapSum() {
        
    }
    
    void insert(string key, int val) {
        int diff = val-m[key].first, n=key.size();
        m[key].first = val;
        for(int i=n-1;i>0;i--){
            m[key.substr(0,i)].second += diff;
        }   
    }
    
    int sum(string prefix) {
        return m[prefix].first + m[prefix].second;
    }
private:
    unordered_map<string, pair<int,int>> m;
};

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

解法二：map<string, int>

class MapSum {
public:
    /** Initialize your data structure here. */
    MapSum() {
        
    }
    
    void insert(string key, int val) {
        m[key] = val;
    }
    
    int sum(string prefix) {
        int n = prefix.size(), res=0;
        for(auto it=m.lower_bound(prefix); it!=m.end();it++){
            if(it->first.substr(0, n)!=prefix) break;
            res += it->second;
        }
        return res;
    }
private:
    map<string, int> m;
};

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