Leetcode.895 最大频率栈

题目链接：

Leetcode.895 最大频率栈 hard

题目描述：

设计一个类似堆栈的数据结构，将元素推入堆栈，并从堆栈中弹出出现频率最高的元素。

实现 $FreqStack$ 类:

• FreqStack() 构造一个空的堆栈。
• void push(int val) 将一个整数 $val$ 压入栈顶。
• int pop() 删除并返回堆栈中出现频率最高的元素。

如果出现频率最高的元素不只一个，则移除并返回最接近栈顶的元素。

示例 1：

输入：
[“FreqStack”,“push”,“push”,“push”,“push”,“push”,“push”,“pop”,“pop”,“pop”,“pop”],
[[],[5],[7],[5],[7],[4],[5],[],[],[],[]]
输出：[null,null,null,null,null,null,null,5,7,5,4]
解释： FreqStack = new FreqStack(); freqStack.push (5);//堆栈为 [5] freqStack.push (7);//堆栈是
[5,7] freqStack.push (5);//堆栈是 [5,7,5] freqStack.push (7);//堆栈是
[5,7,5,7] freqStack.push (4);//堆栈是 [5,7,5,7,4] freqStack.push
(5);//堆栈是 [5,7,5,7,4,5] freqStack.pop ();//返回 5 ，因为 5 出现频率最高。堆栈变成
[5,7,5,7,4]。 freqStack.pop ();//返回 7 ，因为 5 和 7 出现频率最高，但7最接近顶部。堆栈变成
[5,7,5,4]。 freqStack.pop ();//返回 5 ，因为 5 出现频率最高。堆栈变成 [5,7,4]。
freqStack.pop ();//返回 4 ，因为 4, 5 和 7 出现频率最高，但 4 是最接近顶部的。堆栈变成 [5,7]。

提示：

0 <= val <= $10^9$
push 和 pop 的操作数不大于 2 * $10^4$。
输入保证在调用 pop 之前堆栈中至少有一个元素。

解法：哈希表 + 模拟

可以用一个哈希表来记录元素的频率

用一个有序集合（该集合可以按照频率从小到大排序）来记录不同频率的元素。$key$ 即为频率，$value$ 则是一个 $List$（用尾插法，插入不同频率的元素）

Map<Integer,Integer> map ，记录元素的频率

TreeMap<Integer,LinkedList<Integer>> treeMap ， 记录不同频率的元素，频率是按照从小到大的顺序排序的。

图示：

一直模拟这样的过程…

Java代码:

class FreqStack {
    Map<Integer,Integer> map;
    TreeMap<Integer,LinkedList<Integer>> treeMap;
    public FreqStack() {
        map = new HashMap<>();
        treeMap = new TreeMap<>();
    }
    
    public void push(int val) {
        map.put(val,map.getOrDefault(val,0)+1);
        int frequency = map.get(val);

        if(!treeMap.containsKey(frequency)){
            LinkedList<Integer> list = new LinkedList<>();
            treeMap.put(frequency,list);
        }
        treeMap.get(frequency).addLast(val);
    }
    
    public int pop() {
        int key = treeMap.lastKey();
        int val = treeMap.get(key).getLast();

        int cnt = map.get(val);
        cnt--;
        map.put(val,cnt);

        treeMap.get(key).removeLast();
        if(treeMap.get(key).size() == 0){
             treeMap.remove(key);
        }
        return val;
    }
}

/**
 * Your FreqStack object will be instantiated and called as such:
 * FreqStack obj = new FreqStack();
 * obj.push(val);
 * int param_2 = obj.pop();
 */

C++代码:

class FreqStack {
public:
    unordered_map<int,int> cnt;
    map<int,vector<int>> stk;

    FreqStack() {

    }
    
    void push(int val) {
        cnt[val]++;
        stk[cnt[val]].push_back(val);
    }
    
    int pop() {
        //获取指向频率最高的元素集合的指针
        auto it = stk.end();
        it--;

        int ans = it->second.back();
        cnt[ans]--;

        it->second.pop_back();
        if(it->second.empty()) stk.erase(it);
        return ans;
    }

};

/**
 * Your FreqStack object will be instantiated and called as such:
 * FreqStack* obj = new FreqStack();
 * obj->push(val);
 * int param_2 = obj->pop();
 */