C++中的滑动窗口最大值

假设我们有一个名为nums的数组，有一个大小为k的滑动窗口，它从数组的左侧移动到右侧。我们只能看到窗口中的k个数字。每次滑动窗口向右移动一个位置。我们必须找到滑动窗口的最大值。因此，如果输入类似于-[1,3,-1,-3,5,3,6,8]，k为3，则窗口将类似于-

为了解决这个问题，我们将遵循以下步骤：

• 定义一个数组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, ]

Arnab Chakraborty

更新于：2020年5月27日

640 次浏览

开启您的职业生涯

完成课程获得认证

开始学习