Tuyên bố vấn đề
Cho một mảng arr [] và hai số nguyên X và Y. Nhiệm vụ là tìm một mảng con có kích thước ít nhất là X và nhiều nhất là Y với giá trị trung bình lớn nhất
Ví dụ
Nếu mảng đầu vào là {2, 10, 15, 7, 8, 4} và x =3 và Y =3 thì chúng ta có thể nhận được trung bình tối đa 12,5 như sau -
(10 + 15) / 2 = 12.5
Thuật toán
- Lặp lại mọi mảng con có kích thước bắt đầu từ X đến kích thước Y và tìm giá trị trung bình lớn nhất trong số tất cả các mảng con đó.
- Để giảm độ phức tạp về thời gian, chúng ta có thể sử dụng mảng tổng tiền tố để nhận tổng của bất kỳ mảng con nào có độ phức tạp O (1)
Ví dụ
Bây giờ chúng ta hãy xem một ví dụ -
#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; }
Đầu ra
Maximum average = 12.5