代码随想录算法训练营29期Day13|LeetCode 239,347

最新推荐文章于 2024-07-19 22:36:19 发布

tlingyuqi

最新推荐文章于 2024-07-19 22:36:19 发布

阅读量434

点赞数 7

分类专栏：代码随想录算法训练营29期文章标签：算法 leetcode 职场和发展 c++

本文链接：https://blog.csdn.net/tlingyuqi/article/details/135458259

版权

代码随想录算法训练营29期专栏收录该内容

53 篇文章 0 订阅

订阅专栏

文档讲解：

滑动窗口最大值
 前k个高频元素
 栈与队列总结

239.滑动窗口最大值

题目链接：https://leetcode.cn/problems/sliding-window-maximum/submissions/494027196/

思路：

由于我们需要求出的是滑动窗口的最大值，如果当前的滑动窗口中有两个下标 i 和 j，其中 i < j ，并且 i 对应的元素不大于 j 对应的元素，那么会发生什么呢？

当滑动窗口向右移动时，只要 i 还在窗口中，那么 j 一定也还在窗口中，这是 i 在 j 的左侧所保证的。因此，由于 i 对应的元素不大于 j 对应的元素这一条件的存在，nums[i] 一定不会是滑动窗口中的最大值了，我们可以将 nums[i] 永久地移除。

因此我们可以使用一个队列存储所有还没有被移除的下标。当滑动窗口向右移动时，我们需要把一个新的元素放入队列中。为了保持队列的性质，我们会不断地将新的元素与队尾的元素相比较，如果前者大于等于后者，那么队尾的元素就可以被永久地移除，我们将其弹出队列。我们需要不断地进行此项操作，直到队列为空或者新的元素小于队尾的元素。

由于队列中下标对应的元素是严格单调递减的，因此此时队首下标对应的元素就是滑动窗口中的最大值。但与方法一中相同的是，此时的最大值可能在滑动窗口左边界的左侧，并且随着窗口向右移动，它永远不可能出现在滑动窗口中了。因此我们还需要不断从队首弹出元素，直到队首元素在窗口中为止。

核心代码：

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n=nums.size();
        deque<int> q;
        for(int i=0;i<k;i++){
            while (!q.empty()&&nums[i]>=nums[q.back()])  q.pop_back();
            q.push_back(i);
        }
        vector<int> ans={nums[q.front()]};
        for(int i=k;i<n;i++){
            while(!q.empty()&&nums[i]>=nums[q.back()]) q.pop_back();
            q.push_back(i);
            while(q.front()<=i-k) q.pop_front();
            ans.push_back(nums[q.front()]);
        }
        return ans;
    }
};

347.前k个高频元素

题目链接：https://leetcode.cn/problems/top-k-frequent-elements/submissions/494028321/

思路：

这题比上一道简单很多，会上一道就会这道。

前k个高频，无非就是给出现次数排个序嘛。那首先我们得知道出现次数，就需要基于原数组统计一下。由于原数组值的范围很大，因此我们的桶不能直接用数组，就只好用map来统计一下。

统计完之后，基于map的统计结果维护一个size为k的小根堆。在C++里其实就开个优先队列就可以了。枚举每个数字，每次判断堆顶的数字在map中的统计结果是否大于当前数字，大于则不动，小于则将堆顶弹出，将当前数字插入。枚举结束后，留在堆里的k个数字就是所有数字中的前k个高频元素了。

核心代码：

class Solution {
public:
    static bool cmp(pair<int,int>& a,pair<int,int>& b){
        return a.second>b.second;
    }

    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> cnt;
        for(auto& v:nums) cnt[v]++;
        priority_queue<pair<int,int>,vector<pair<int, int>>,decltype(&cmp)> q(cmp);
        for(auto& [num,count]:cnt){
            if(q.size()==k){
                if(q.top().second<count){
                    q.pop();
                    q.emplace(num,count);
                }
            }
            else{
                q.emplace(num, count);
            }
        }
        vector<int> ans;
        while (!q.empty()){
            ans.emplace_back(q.top().first);
            q.pop();
        }
        return ans;
    }
};