【算法训练 day13 滑动窗口的最大值、前k个高频元素】

---

---

一、滑动窗口的最大值-LeetCode 239

Leecode链接: leetcode 239
文章链接: 代码随想录
视频链接: B站

给定一个数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值。

示例1：

输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
输出：[3,3,5,5,6,7]

[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7

思路

使用一个单调队列，该队列存放窗口中可能是最大值的数据，每次移动时，弹出数据，插入时需要弹出比当前插入值小得所有数，这样最大值永远处于队列的front()。

实现代码

1.左闭右闭

//cpp
class Solution {
private:
    class MyQueue { //单调队列（从大到小）
    public:
        deque<int> que; // 使用deque来实现单调队列
        // 每次弹出的时候，比较当前要弹出的数值是否等于队列出口元素的数值，如果相等则弹出。
        // 同时pop之前判断队列当前是否为空。
        void pop(int value) {
            if (!que.empty() && value == que.front()) {
                que.pop_front();
            }
        }
        // 如果push的数值大于入口元素的数值，那么就将队列后端的数值弹出，直到push的数值小于等于队列入口元素的数值为止。
        // 这样就保持了队列里的数值是单调从大到小的了。
        void push(int value) {
            while (!que.empty() && value > que.back()) {
                que.pop_back();
            }
            que.push_back(value);

        }
        // 查询当前队列里的最大值 直接返回队列前端也就是front就可以了。
        int front() {
            return que.front();
        }
    };
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MyQueue que;
        vector<int> result;
        for (int i = 0; i < k; i++) { // 先将前k的元素放进队列
            que.push(nums[i]);
        }
        result.push_back(que.front()); // result 记录前k的元素的最大值
        for (int i = k; i < nums.size(); i++) {
            que.pop(nums[i - k]); // 滑动窗口移除最前面元素
            que.push(nums[i]); // 滑动窗口前加入最后面的元素
            result.push_back(que.front()); // 记录对应的最大值
        }
        return result;
    }
};

问题

一刷时跳过了，没有思路，相当于学习新的知识：单调队列。

---

二、前k个高频元素-LeetCode 347

Leecode链接: LeetCode 347
文章链接: 代码随想录
视频链接: B站

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例1：

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

思路

使用map记录对应元素以及它的数量，然后对数量进行排序，这里使用了堆排序。

实现代码

//cpp
class Solution {
public:
    class mycomaprison{
    public:
        bool operator()(pair<int,int>& i,pair<int,int>& j) const{
            return i.second>j.second;
        }
    };

    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int>m;//存放数据以及数据出现的频率
        vector<int>res(k);//存放结果

        //元素计数
        for(int i = 0;i<nums.size();i++){
            m[nums[i]]++;
        }

        priority_queue<pair<int,int>,vector<pair<int,int>>,mycomaprison> min_head;//建立小顶堆

        for(auto it = m.begin();it != m.end();it++){
            min_head.push(*it);

            if(min_head.size()>k){
                min_head.pop();
            }
        }

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

        return res;
    }
};

总结

一刷跳过，学习使用仿函数的使用以及容器适配器的使用。