最大频率栈
难度:
标签:
题目描述
代码结果
运行时间: 184 ms, 内存: 24.8 MB
/*
思路:
1. 使用两个HashMap来记录元素的频率(freqMap)和频率对应的元素栈(freqStackMap)。
2. push操作:更新元素的频率,并将元素添加到对应频率的栈中。
3. pop操作:找到最高频率的栈,弹出栈顶元素并更新频率。
4. 使用Java Stream在操作集合时更简洁。
*/
import java.util.HashMap;
import java.util.Map;
import java.util.Stack;
public class FreqStack {
private Map<Integer, Integer> freqMap;
private Map<Integer, Stack<Integer>> freqStackMap;
private int maxFreq;
public FreqStack() {
freqMap = new HashMap<>();
freqStackMap = new HashMap<>();
maxFreq = 0;
}
public void push(int val) {
int freq = freqMap.getOrDefault(val, 0) + 1;
freqMap.put(val, freq);
if (freq > maxFreq) {
maxFreq = freq;
}
freqStackMap.computeIfAbsent(freq, k -> new Stack<>()).push(val);
}
public int pop() {
Stack<Integer> stack = freqStackMap.get(maxFreq);
int val = stack.pop();
if (stack.isEmpty()) {
maxFreq--;
}
freqMap.put(val, freqMap.get(val) - 1);
return val;
}
}
解释
方法:
这个题解的思路是使用两个主要的数据结构:一个字典用来存储每个元素的出现频率,一个列表用来存储每个频率对应的元素集合。每次元素入栈时,更新该元素的频率,并将元素添加到对应频率的集合中。出栈时,从最高频率的集合中弹出元素,如果该集合为空,则移除该集合。
时间复杂度:
O(1)
空间复杂度:
O(n)
代码细节讲解
🦆
在实现FreqStack类的时候,你是如何确保pop操作始终返回最近入栈的出现频率最高的元素的?
▷🦆
你提到使用字典来存储每个元素的出现频率,这种方法在所有情况下都能准确追踪频率变化吗,即使是在频繁的push和pop操作之后?
▷🦆
为什么选择使用列表来存储每个频率对应的元素集合而不是其他数据结构,比如堆或者平衡树?
▷🦆
在pop操作中,如果最高频率的集合为空,你会如何处理并确保操作的正确性?
▷