增、删、最大最小O(1)数据结构-算法题

题意：请你设计一个用于存储字符串计数的数据结构，并能够返回计数最小和最大的字符串。

传送门

实现 AllOne 类：

• AllOne() 初始化数据结构的对象。
• inc(String key) 字符串 key 的计数增加 1 。如果数据结构中尚不存在 key ，那么插入计数为 1 的 key 。
• dec(String key) 字符串 key 的计数减少 1 。如果 key 的计数在减少后为 0 ，那么需要将这个 key 从数据结构中删除。测试用例保证：在减少计数前，key 存在于数据结构中。
• getMaxKey() 返回任意一个计数最大的字符串。如果没有元素存在，返回一个空字符串 “” 。
• getMinKey() 返回任意一个计数最小的字符串。如果没有元素存在，返回一个空字符串 “” 。

输入
[“AllOne”, “inc”, “inc”, “getMaxKey”, “getMinKey”, “inc”, “getMaxKey”, “getMinKey”]
[[], [“hello”], [“hello”], [], [], [“leet”], [], []]
输出
[null, null, null, “hello”, “hello”, null, “hello”, “leet”]
解释
AllOne allOne = new AllOne();
allOne.inc(“hello”);
allOne.inc(“hello”);
allOne.getMaxKey(); // 返回 “hello”
allOne.getMinKey(); // 返回 “hello”
allOne.inc(“leet”);
allOne.getMaxKey(); // 返回 “hello”
allOne.getMinKey(); // 返回 “leet”

题解：利用map和set，map用来计数，set用来更新对应key的出现次数

class AllOne {
public:
    struct node{
        string key;
        int val;
        bool friend operator < (const node &a, const node &b) {
            if(a.val == b.val) {
                return a.key < b.key;
            }
            return a.val > b.val;
        }
    };

    set<node> s;
    unordered_map<string, int> bk;

    AllOne() {
        s.clear();
        bk.clear();
    }
    
    void inc(string key) {
        if(!bk.count(key)) {
            bk[key] = 1;
        } else {
            s.erase(node{key, bk[key]++});
        }
        s.insert(node{key, bk[key]});
    }
    
    void dec(string key) {
        s.erase(node{key, bk[key]--});
        if(bk[key] == 0) {
            bk.erase(key);
        } else {
            s.insert(node{key, bk[key]});
        }
    }
    
    string getMaxKey() {
        return s.empty() ? "" : s.begin()->key;
    }
    
    string getMinKey() {
        return s.empty() ? "" : s.rbegin()->key;
    }
};