一、滑动窗口的最大值
题目描述
面试题59 - I. 给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
思路分析
使用双端单调队列(deque)的内在逻辑:
- 一旦出现比之前更大的数,则最大值更新为这个数,且之前的数不可能再成为最大数(队列的先进先出原则),则不需保存之前的数;
对应操作:小于当前数的数全部出双端单调队列,当前数入双端单调队列。 - 出现比之前小的数,在当前最大值出队后,有可能成为最大值,需要保存。
对应操作:当前数直接入双端单调队列。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
int len = nums.size();
vector<int> res;
if(len<1 || k < 1)
return res;
// 初始化
deque<int> dq; // 存放的是index
for(int i=0;i<k-1;i++){
// 初始化时不需要判断队首不在窗口内
while(!dq.empty()&&nums[i]>nums[dq.back()])
dq.pop_back();
dq.push_back(i); //从队尾插入到队列中
}
for(int i=k-1;i<len;i++){ // 放入第k个值,就可以开始输入结果了
// 检查队首是否在窗口内
if(!dq.empty() && (dq.front()<i-k+1))
dq.pop_front();
//将新进数字和队尾端元素进行判断 小于的话直接进队
while(!dq.empty()&&nums[i]>nums[dq.back()])
dq.pop_back();
dq.push_back(i);
res.push_back(nums[dq.front()]); //vector存放每个窗口内的最大值
}
return res;
}
};
二、队列的最大值
题目描述
面试题59-2:请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_value、push_back 和 pop_front 的均摊时间复杂度都是O(1)。
若队列为空,pop_front 和 max_value 需要返回 -1
参考:包含MIN函数的栈
思路分析
此时队列q保持了输入顺序,双端队列d是可能出现的最大值的递减序列,说明如下
-
若这里输入11321,则d中保存的为321,q中保存的为11321。出队顺序只能为11321。
假设已经出队11,此时max=d.front()=3,d={3,2,1},q={3,2,1},
再出队3,d={2,1},q={2,1},d的队首元素2依然是当前最大
再出队2,d={1},q={1},d的队首元素1依然是当前最大 -
若输入11323,则d中保存的为33,q中保存的为11323。出队顺序只能为11323。
假设已经出队11,此时max=d.front()=3,d={3,3},q={3,2,3},
再出队3,d={3},q={2,3},d的队首元素3依然是当前最大
再出队2,d={3},q={3},d的队首元素3依然是当前最大,可知2并没有成为最大值的机会,d中存放的都是可能出现的最大值
class MaxQueue {
private:
deque<int> d; //d中存放的都是可能出现的最大值(递减),双端队列队首元素即是当前最大值
queue<int> q; //用队列q保持了输入顺序
public:
MaxQueue() {}
int max_value() {
if (d.empty())
return -1;
return d.front();
}
void push_back(int value) {
while (!d.empty() && d.back() < value)
d.pop_back();
d.push_back(value);
q.push(value);
}
int pop_front() { //pop函数返回不为空
if (q.empty())
return -1;
int ans = q.front();
if (d.front() == ans)
d.pop_front();
q.pop();
return ans;
}
};