题目:给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。返回滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
思路:定义一个单调递减队列表示滑动窗口,单调队列里不用维护滑动窗口里的所有值。
当滑动窗口移动时,首先要移除窗口里的第一个元素,但因为是用单调递减队列去维护数据,所以数据移出时,首先判断要移除的元素是否与队列里的第一个元素相等,相等时,队列弹出第一个元素。不相等则说明该数据在新的滑动窗口里,不用弹出。
然后滑动窗口移入新的数据,新的元素与队列尾部元素比大小,新元素大于队尾数据,将队尾数据删除,重复执行这一操作,直到新元素小于队尾元素,然后将新元素插入队列。
每次滑动窗口移动时,队列里的第一个元素就是当前滑动窗口的最大值。
代码:
class Solution { //滑动窗口最大值 单调队列
public:
class MonotonousQueue { //单调递减队列
public:
deque<int> que;
void push(int val) {
while (!que.empty() && que.back() < val) {
que.pop_back();
}
que.push_back(val);
}
void pop(int val) {
if (!que.empty() && que.front() == val) {
que.pop_front();
}
}
int front() {
return que.front();
}
};
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> result;
MonotonousQueue que;
for (int i = 0; i < k; ++i) {
que.push(nums[i]);
}
result.push_back(que.front());
for (int i = k; i < nums.size(); ++i) {
que.pop(nums[i - k]);
que.push(nums[i]);
result.push_back(que.front());
}
return result;
}
};
int main() {
vector<int> nums = { 1, 3, -1, -3, 5, 3, 6, 7 };
int k = 3;
Solution s;
vector<int> result = s.maxSlidingWindow(nums, k);
for (int i : result) {
cout << i << "\t";
}
cout << endl;
return 0;
}
参考资料:代码随想录