力扣(LeetCode)895. 最大频率栈(C++)

最新推荐文章于 2024-06-09 18:25:10 发布

清墨韵染

最新推荐文章于 2024-06-09 18:25:10 发布

阅读量627

点赞数 2

分类专栏：墨染leetcode 文章标签： c++ leetcode 算法哈希容器

本文链接：https://blog.csdn.net/Innocence02/article/details/128110180

版权

墨染leetcode 专栏收录该内容

138 篇文章 2 订阅

订阅专栏

本文介绍了一种数据结构设计，通过使用unordered_map维护每个数的出现次数，并结合栈来跟踪最大频率。压栈和弹栈操作优化为O(1)，确保了时间复杂度为线性且空间复杂度达到n。主要关注频率统计与栈操作的高效实现，适用于实时频率跟踪应用。

摘要由CSDN通过智能技术生成

设计

①维护最大频率，②维护每个数的出现次数，③维护出现次数对应的栈。

压栈时，新数压入出现次数对应的栈，每次压入新数，维护最大频率(所有出现次数中的最大出现次数)。

弹栈时，找最大频率对应的栈，取出栈顶元素即可。弹栈操作可能使栈为空，栈空说明没有最大频率的数了，最大频率 $- 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 (1)$ ， $n$ 次操作的时间复杂度 $O (n)$ 。
空间复杂度 : $O (n)$ ，频率栈的最坏空间复杂度 $O (n)$ 。