【LeetCode】895. 最大频率栈

895. 最大频率栈

知识点：栈；哈希表

题目描述

实现 FreqStack，模拟类似栈的数据结构的操作的一个类。

FreqStack 有两个函数：

push(int x)，将整数 x 推入栈中。
pop()，它移除并返回栈中出现最频繁的元素。
如果最频繁的元素不只一个，则移除并返回最接近栈顶的元素。

示例

输入：
["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]
解释：
执行六次 .push 操作后，栈自底向上为 [5,7,5,7,4,5]。然后：

pop() -> 返回 5，因为 5 是出现频率最高的。
栈变成 [5,7,5,7,4]。

pop() -> 返回 7，因为 5 和 7 都是频率最高的，但 7 最接近栈顶。
栈变成 [5,7,5,4]。

pop() -> 返回 5 。
栈变成 [5,7,4]。

pop() -> 返回 4 。
栈变成 [5,7]。

---

解法一：哈希表

这个题目需要关注两个点：

• 频率表：对于每个元素，我们要统计出其出现的频率，这自然的就会想到哈希表；用一个频率哈希表存储val和对应的freq。
• 对于频率相同的，要按照离顶近的先出，也就是说后入的先出，所以可以再来一个哈希表，key值是频率，value是一个栈，栈里存放着元素，所以每次弹出的时候都是弹出最大频率的栈顶。
• 注意一个点，就是这个maxFreq一定是连续的，因为在入栈的时候，对于在最大频率处的元素，其也一定存在于比它小的栈里，都是逐步增大才到最大频率栈里的。

class FreqStack {
        int maxFreq = 0;
        HashMap<Integer, Integer> valFreq;
        HashMap<Integer, Stack<Integer>> sameFreqVal;
    public FreqStack() {
        valFreq = new HashMap<>();
        sameFreqVal = new HashMap<>();
    }
    
    public void push(int val) {
        int freq = valFreq.getOrDefault(val, 0)+1;  //更新频率表；
        valFreq.put(val, freq);

        sameFreqVal.putIfAbsent(freq, new Stack<Integer>());  //if有key，就不添加，没有key的话就添加；
        sameFreqVal.get(freq).push(val);

        maxFreq = Math.max(freq, maxFreq);
    }
    
    public int pop() {
        Stack<Integer> temp = sameFreqVal.get(maxFreq);  //最大频率对应的元素；

        int top = temp.pop();
        if(temp.isEmpty()){   //弹出后当前栈为空，那就把这条记录删除；
            sameFreqVal.remove(maxFreq);
            maxFreq--;   //频率一定是连续的，因为if一个数在最大频率栈，那在比它频率小的栈里也一定有；
        }
        int freq = valFreq.get(top);
        if(freq-- == 0){
            valFreq.remove(top);
        }else{
            valFreq.put(top, freq--);  //更新频率表；
        }
        return top;
    }
}

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