LeetCode #895. 最大频率栈 题解 C/C++

主要思路

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;
    }
};