C++中的滑動視窗最大值


假設我們有一個名為nums的陣列,有一個大小為k的滑動視窗,它從陣列的左側移動到右側。我們只能看到視窗中的k個數字。每次滑動視窗向右移動一個位置。我們必須找到滑動視窗的最大值。因此,如果輸入類似於-[1,3,-1,-3,5,3,6,8],k為3,則視窗將類似於-

視窗位置最大值
13-1-353683
13-1-353683
13-1-353683
13-1-353685
13-1-353686
13-1-353688

為了解決這個問題,我們將遵循以下步驟:

  • 定義一個數組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, ]

更新於:2020年5月27日

640 次瀏覽

開啟您的職業生涯

完成課程獲得認證

開始學習
廣告