LeetCode日记（最大频率栈）第十天

题目来自LeetCode编号895题。题目如下：

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

实现 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]。

思路如下：

我们用map集合来储存频率，再建立一个集合储存不同频率的栈，让储存时根据频率，让其进入相应频率的栈，取出的时候我们从频率高的栈开始取出。并且更新频率为下次提取做准备，如果栈空了我们就把栈从集合中删除。

代码如下：

class FreqStack {
    private final Map<Integer, Integer> cnt = new HashMap<>();
    private final List<Deque<Integer>> stacks = new ArrayList<>();
    public FreqStack() {
        
    }
    
    public void push(int val) {
    int c=cnt.getOrDefault(val,0);//获取这个数目前的频率
    if(c==stacks.size())//如果这个频率等于目前的最大频率，我们就为其开辟一个新的栈存储
    {
        stacks.add(new ArrayDeque<>());
    }
    stacks.get(c).push(val);//并将这个数进入新栈
    cnt.put(val,c+1);//存入后，更新目前的这个数的频率
    }
    
    public int pop() {
    int len=stacks.size()-1;//获取集合最大频率，我们从最后那个栈取元素，即目前频率最大的元素栈
    int i=stacks(len).pop();
    if(stacks(len).isEmpty())//取出元素后栈为空我们就删去。
    {
        stacks.remove(len);
    }
    cnt.put(val,get(val)-1);//更新这个数的频率
    return i;//返回我们出栈的数。
    }
}