代码随想录day13

原创已于 2023-07-17 22:43:15 修改 · 112 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #职场和发展

于 2023-07-10 21:13:43 首次发布

239滑动窗口最大值

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

双端队列相当惊艳

增队列队尾到对头单增

i-k+1所在的if语句指的是当我的窗口要滑出最大值所在的区域时，不得不把队头删掉

347 前k个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k)  {
        typedef pair<int,int> P;
        unordered_map<int,int>map;
        for(int i=0;i<nums.size();i++)
        {
            map[nums[i]]++; // 数字 数量
        }
        auto f=[](auto a,auto b){return a.second>b.second;};
        priority_queue<P,vector<P>,decltype(f)>pq(f);
        for(auto a:map)
        {
            pq.push(a);
            if (pq.size()>k)pq.pop();
        }
        vector<int> result(k);
        for (int i = k - 1; i >= 0; i--) 
        {
            result[i] = pq.top().first;
            pq.pop();
        }
        return result;
    }
};

学习到了优先级队列怎么放入map类型并且可以利用pq.size()进行剪枝