剑指 Offer 59 - I. 滑动窗口的最大值
给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: 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
思路
利用deque构建一个单调递增的队列,每次front都是当前滑动窗口的最大值;每次push_back,保证当前元素不大于队尾元素,否则不断pop_back,直到当前元素不大于队尾元素则push_back;每次移除滑动窗口的末尾值时和当前队首值比较,如果删除的值正好时队首的值,则pop_front,否则不操作;
//构建单调递增队列,保证每次队首的元素都是当前窗口的最大值
class MyQueue{
deque<int> que;
public:
//当队列不为空且当前值不大于队尾值则入队列,队尾元素出队列直到当前值不大于队尾值
void push(int value){
while(!que.empty() && que.back() < value){
que.pop_back();
}
//当前值不大于队尾值,当前值入队列
que.push_back(value);
}
//当滑动窗口要移除的元素等于当前队首元素时,队首元素出队列(移除元素是否是当前窗口的最大值,是则将队首元素移除)
//否则移除的元素不是当前窗口的最大值,则队列的元素不动
void pop(int value){
if(!que.empty() && que.front() == value)
que.pop_front();
}
int front(){
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> result;
if(k == 0)
return result;
//将前k个数入队列
for(int i = 0; i < k; ++i)
que.push(nums[i]);
result.push_back(que.front());
//滑动窗口,先pop再push
for(int i = k; i < nums.size(); ++i){
que.pop(nums[i - k]);
que.push(nums[i]);
result.push_back(que.front());
}
return result;
}