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

题目

• 题目链接，代码
• 题目链接，代码

初见思路

239. 滑动窗口最大值

主要根据代码随想录上的思路来的，使用deque来实现一个我们自己的单调队列。

347.前 K 个高频元素

用map来保存出现的数字和对应的次数。用一个priority_queue （max_heap）来对数字的出现次数进行排序，pop掉前K个就行了。

更新思路

239. 滑动窗口最大值

流程：

1. 将开头的k个值push存入窗口中（单调队列）。push的时候会将需要push的值与入口（back）处的值进行比较，如果需要push的更大那就移除入口处的那个值，以此保证front的值一直都是当前窗口最大的。储存最大值（front）。
2. 开始移动窗口，pop掉窗口最左边的值（如果这个值是前一个窗口最大值的话，如果不是就没必要pop，之后push的时候会pop掉），push新的值（同样也要pop掉比较小的值），然后将front存入result。
3. 重复第二步。

class MyQueue {
public:
    deque<int> que;

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

    void push(int value){
        while(!que.empty() && value > que.back()){
            que.pop_back(); // remove smaller values
        }
        que.push_back(value);
    }

    int front(){
        return que.front();
    }
};

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MyQueue q;
        vector<int> res;

        // first window
        for(int i = 0; i < k; ++i){
            q.push(nums[i]);
        }

        res.push_back(q.front());
        for(int i = k; i < nums.size(); ++i){
            q.pop(nums[i - k]); // pop the slided value
            q.push(nums[i]); // push to back, pop smaller values
            res.push_back(q.front()); // max val for this sliding window.

        }
        return res;
    }
};

347.前 K 个高频元素

max heap父节点比子节点大，在max heap中储存一个pair，根据map中的次数进行排序。pop root添加进result中。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> map;
        for(int n : nums){
            ++map[n];
        }

        vector<int> res;
        // pair<first, second>: first is frequency,  second is number
        priority_queue<pair<int, int>> pq;

        for(auto it = map.begin(); it != map.end(); it++){
            pq.push(make_pair(it->second, it->first));
            if(pq.size() > (int)map.size() - k){ //k frequent
                res.push_back(pq.top().second);
                pq.pop();
            }
        }
        return res;
    }
};

今日总结

学习时长

2h

收获

https://programmercarl.com/%E6%A0%88%E4%B8%8E%E9%98%9F%E5%88%97%E6%80%BB%E7%BB%93.html