以 C++ 中至少 X 和最多 Y 大小的子陣列為最大平均值


問題陳述

給定一個數組 arr[] 和兩個整數 X 和 Y。任務是用至少 X 和最多 Y 的大小找到具有最大平均值的一個子陣列

示例

如果輸入陣列為 {2, 10, 15, 7, 8, 4},x = 3 且 Y = 3,則我們可以得到最大平均值 12.5,如下所示−

(10 + 15) / 2 = 12.5

演算法

  • 從大小從 X 開始到 Y 大小的每個子陣列進行迭代,並在所有此類子陣列中找到最大平均值。
  • 為了降低時間複雜度,我們可以使用字首和陣列以 O(1) 的複雜度獲取任何子陣列的和

示例

現在讓我們看一個例子−

 線上演示

#include <bits/stdc++.h>
using namespace std;
double getMaxAverage(int *arr, int n, int x, int y) {
   int prefix[n];
   prefix[0] = arr[0];
   for (int i = 1; i < n; ++i) {
      prefix[i] = prefix[i - 1] + arr[i];
   }
   double maxAverage = 0;
   for (int i = 0; i < n; ++i) {
      for (int j = i + x - 1; j < i + y && j < n; ++j) {
         double sum = prefix[j];
         if (i > 0) {
            sum = sum - prefix[i - 1];
            double current = sum / double(j - i + 1);
            maxAverage = max(maxAverage,current);
         }
      }
   }
   return maxAverage;
}
int main() {
   int arr[] = {2, 10, 15, 7, 8, 4};
   int x = 2;
   int y = 3;
   int n = sizeof(arr) / sizeof(arr[0]);
   cout << "Maximum average = " << getMaxAverage(arr, n, x, y) << endl;
   return 0;
}

輸出

Maximum average = 12.5

更新於: 2019-12-31

187 次瀏覽

啟動您的 職業生涯

完成課程並獲得認證

開始
廣告
© . All rights reserved.