Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

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

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • 배열 정의

  • 하나의 이중 종료 대기열 dq 정의

  • 숫자의 크기가 0과 같으면 다음을 반환합니다.

  • i를 초기화하기 위해 :=0, i

    • 동안 dq는 비어 있지 않고 nums[dq의 마지막 요소] ​​

      • dq의 마지막 요소 삭제

    • dq 끝에 i 삽입

  • i를 초기화하기 위해 :=k, i

    • ans

      에 (nums[front element of dq])를 삽입합니다.
    • dq가 비어 있지 않고 dq <(i-k + 1)의 앞 요소인 동안 −

      • dq에서 앞 요소 삭제

    • dq가 비어 있지 않고 nums[dq의 마지막 요소] ​​

      • dq의 마지막 요소 삭제

    • dq 끝에 i 삽입

  • nums[dq의 앞 요소]를 끝에 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, ]