设计
①维护最大频率,②维护每个数的出现次数,③维护出现次数对应的栈。
压栈时,新数压入出现次数对应的栈,每次压入新数,维护最大频率(所有出现次数中的最大出现次数)。
弹栈时,找最大频率对应的栈,取出栈顶元素即可。弹栈操作可能使栈为空,栈空说明没有最大频率的数了,最大频率 − 1 -1 −1 即可。
class FreqStack {
public:
unordered_map<int,int> cnt;//统计出现次数
unordered_map<int,vector<int>> stk;//次数对应栈
int freq;
FreqStack() {}
void push(int val) {
stk[++cnt[val]].push_back(val);
freq = max(freq,cnt[val]);
}
int pop() {
int val = stk[freq].back();
stk[freq].pop_back();
cnt[val]--;
if(stk[freq].empty()) freq--;
return val;
}
};
- 时间复杂度 : O ( n ) O(n) O(n) , 每个操作的时间复杂度 O ( 1 ) O(1) O(1) , n n n 次操作的时间复杂度 O ( n ) O(n) O(n) 。
- 空间复杂度 : O ( n ) O(n) O(n) , 频率栈的最坏空间复杂度 O ( n ) O(n) O(n) 。