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

C++ STL에서 set을 사용하는 k 크기의 모든 하위 배열의 최대값

<시간/>

이 자습서에서는 C++ STL에서 set을 사용하여 k 크기의 모든 하위 배열을 최대로 가져오는 프로그램에 대해 설명합니다.

이를 위해 크기가 N이고 정수 K인 배열이 제공됩니다. 우리의 임무는 각 K개의 요소에서 최대 요소를 가져와 더하고 인쇄하는 것입니다.

예시

#include <bits/stdc++.h>
using namespace std;
//returning sum of maximum elements
int maxOfSubarrays(int arr[], int n, int k){
   set<pair<int, int> > q;
   set<pair<int, int> >::reverse_iterator it;
   //inserting elements
   for (int i = 0; i < k; i++) {
      q.insert(pair<int, int>(arr[i], i));
   }
   int sum = 0;
   for (int j = 0; j < n - k + 1; j++) {
      it = q.rbegin();
      sum += it->first;
      q.erase(pair<int, int>(arr[j], j));
      q.insert(pair<int, int>(arr[j + k], j + k));
   }
   return sum;
}
int main(){
   int arr[] = { 4, 10, 54, 11, 8, 7, 9 };
   int K = 3;
   int n = sizeof(arr) / sizeof(arr[0]);
   cout << maxOfSubarrays(arr, n, K);
   return 0;
}

출력

182