使用 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

更新於: 2020 年 4 月 6 日

192 次檢視

開啟你的職業生涯

完成課程,獲取認證

入門
廣告
© . All rights reserved.