今天的内容是关于队列稍微复杂一些的应用
第一题、LeetCode239 滑动窗口最大值 https://leetcode.cn/problems/sliding-window-maximum/
使用队列来解决这一道题,首先遇到的问题是怎么把知道队列里的最大值,我们需要保证能把最大值放在队列出口处才能知道最大值。因此每一个push进去新的元素,都要和前面的元素数值进行比较。
注意:如果push的数值大于入口元素的数值,那么就将队列后端的数值弹出,直到push的数值小于等于队列入口元素的数值为止。 这样就保持了队列里的数值是单调从大到小的了。
而pop则要比较当前要弹出的数值是否等于队列出口元素的数值,如果相等则弹出这样就保证了在后面窗口往后滑动时,不会pop错元素。
class Solution {
private:
class Myqueue{
public:
deque<int> que;
// 每次弹出的时候,比较当前要弹出的数值是否等于队列出口元素的数值,如果相等则弹出。
// 同时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);
}
int front(){
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
Myqueue que;
vector<int> res;
for(int i = 0; i < k; i++){
que.push(nums[i]);
}
res.push_back(que.front()); // result 记录前k的元素的最大值
for(int i = k; i < nums.size(); i++){
que.pop(nums[i - k]); //先把出口的元素pop了,注意若该元素已经被卷走就不会被pop
que.push(nums[i]);
res.push_back(que.front());
}
return res;
}
};
第二题、LeetCode 347、前k个高频元素 https://leetcode.cn/problems/top-k-frequent-elements/