C++中的滑动窗口最大值
假设我们有一个名为nums的数组,有一个大小为k的滑动窗口,它从数组的左侧移动到右侧。我们只能看到窗口中的k个数字。每次滑动窗口向右移动一个位置。我们必须找到滑动窗口的最大值。因此,如果输入类似于-[1,3,-1,-3,5,3,6,8],k为3,则窗口将类似于-
窗口位置 | 最大值 | |||||||
---|---|---|---|---|---|---|---|---|
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 3 |
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 3 |
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 3 |
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 5 |
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 6 |
1 | 3 | -1 | -3 | 5 | 3 | 6 | 8 | 8 |
为了解决这个问题,我们将遵循以下步骤:
定义一个数组ans
定义一个双端队列dq
如果nums的大小为0,则返回ans
初始化i := 0,当i<k时,i加1,执行:
当dq不为空且nums[dq的最后一个元素] < nums[i]时,执行:
删除dq的最后一个元素
在dq的末尾插入i
初始化i := k,当i<nums.size()时,i加1,执行:
将(nums[dq的第一个元素])插入ans
当dq不为空且dq的第一个元素 < (i-k + 1)时,执行:
删除dq的第一个元素
当dq不为空且nums[dq的最后一个元素] < nums[i]时,执行:
删除dq的最后一个元素
在dq的末尾插入i
将nums[dq的第一个元素]插入ans的末尾
返回ans
示例
让我们看看下面的实现,以便更好地理解:
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } void print_vector(vector<vector<auto> > v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << "["; for(int j = 0; j <v[i].size(); j++){ cout << v[i][j] << ", "; } cout << "],"; } cout << "]"<<endl; } class Solution { public: vector<int> maxSlidingWindow(vector<int>& nums, int k) { vector <int> ans; deque <int> dq; if(nums.size()==0)return ans; for(int i =0;i<k;i++){ while(!dq.empty() && nums[dq.back()]<nums[i])dq.pop_back(); dq.push_back(i); } for(int i = k;i<nums.size();i++){ ans.push_back(nums[dq.front()]); while(!dq.empty() && dq.front()<(i-k + 1))dq.pop_front(); while(!dq.empty() && nums[dq.back()]<nums[i])dq.pop_back(); dq.push_back(i); } ans.push_back(nums[dq.front()]); return ans; } }; main(){ Solution ob; vector<int> v = {1,3,-1,-3,5,3,6,8}; print_vector(ob.maxSlidingWindow(v,3)); }
输入
{1,3,-1,-3,5,3,6,8}
输出
[3, 3, 5, 5, 6, 8, ]
广告