代码随想录算法训练营第十三天| 239. 滑动窗口最大值、347.前 K 个高频元素

LeetCode 239 滑动窗口最大值

题目链接：239
思路：需要求出滑动窗口的最大值，因此只需要维护窗口中可能成为最大值的元素即可；创建一个单调队列，即队列中元素降序排列，即队头为最大元素，队尾为最小元素。当滑动队列时，若需要弹出的元素等于队头元素，说明当前的最大值不在新的窗口中，因此弹出队头，否则由于需要弹出的元素不是最大值，因此弹出与否不影响当前窗口的最大值，所以不操作。需要插入的元素需与队尾元素挨个比较，将小于该元素的队尾元素依次弹出，然后插入该元素，因为小于当前元素的队尾元素比当前元素早插入，因此本该早弹出，而它们的值小于当前元素，因此在他们的生命周期内不可能成为最大元素，所以无需维护，弹出即可，从而保证队列的单调性。
代码：

class Solution {
public:

    class MonoQueue {
    public:
        MonoQueue() { }

        void pop(int x) {
            if (!que_.empty() && x == que_.front()) {
                que_.pop_front();
            }
        }

        void push(int x) {
            while (!que_.empty() && que_.back() < x) {
                que_.pop_back();
            }
            que_.push_back(x);
        }

        int& front() {
            return que_.front();
        }

    private:
        deque<int> que_;
    };

    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MonoQueue slide;
        vector<int> res;

        for (int i=0; i<k; ++i) {
            slide.push(nums[i]);
        }
        res.push_back(slide.front());

        for (int i=0,j=k;j<nums.size(); ++i,++j) {
            slide.pop(nums[i]);
            slide.push(nums[j]);
            res.push_back(slide.front());
        }

        return res;
    }
};

LeetCode 347 前 K 个高频元素

题目链接：347
思路：首先通过一个哈希表存储数组中每个元素的频率；其次，需要其中前k个高频元素，因此可以构建一个小顶堆，当堆元素小于k时，直接插入；当堆元素等于k时，若当前元素大于堆顶，则弹出堆顶并插入元素，否则舍弃。
代码：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        for (int i=0; i<nums.size(); ++i) {
            mp[nums[i]]++;
        }

        auto cmp = [](const pair<int,int>& a, const pair<int,int>& b) -> bool {
                            return a.second > b.second;
                    }; 
        priority_queue<pair<int,int>, vector<pair<int,int>>, decltype(cmp)> que(cmp);
        for (unordered_map<int, int>::iterator it=mp.begin(); it!=mp.end(); ++it) {
            que.push(*it);
            if (que.size() > k) {
                que.pop();
            }
        }

        vector<int> res(k);
        for (int i=0; i<k; ++i) {
            res[i] = que.top().first;
            que.pop();
        }

        return res;
    }
};