主要思路
1.维护一个map映射,map<int,int>freq;为元素x到该元素出现的频率;即key为元素x,value为该元素出现的次数。
2.在维护一个 频率到具有该频率元素集合的映射,map<int,stack>group;即key值为频率,value为一个栈,其中存放具有该频率的元素,这样就满足了“如果最频繁的元素不只一个,则移除并返回最接近栈顶的元素”这一要求。
3.最后始终维护一个全局变量max_freq记录最大频率.
class FreqStack {
private:
map<int,int> freq;//x 到 x出现次数的映射
map<int ,stack<int>> group;//频率到具有该频率的映射
int max_freq;//最大频率
public:
FreqStack() {
max_freq = 0;
}
void push(int val) {
int time = ++freq[val];//记录当前val值的频率
max_freq = max(max_freq,time);//更新最大的频率
group[time].push(val);//栈中存放具有该频率(time)的元素
}
int pop() {
int ans = group[max_freq].top();//取出频率最高的元素
group[max_freq].pop();//将其删除
freq[ans]--;//该元素的频率-1
//更新最大频率max_freq
if(group[max_freq].size()==0){max_freq--;}
return ans;
}
};