设计
①维护最大频率,②维护每个数的出现次数,③维护出现次数对应的栈。
压栈时,新数压入出现次数对应的栈,每次压入新数,维护最大频率(所有出现次数中的最大出现次数)。
弹栈时,找最大频率对应的栈,取出栈顶元素即可。弹栈操作可能使栈为空,栈空说明没有最大频率的数了,最大频率 −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) , nnn 次操作的时间复杂度 O(n)O(n)O(n) 。
- 空间复杂度 : O(n)O(n)O(n) , 频率栈的最坏空间复杂度 O(n)O(n)O(n) 。