문제 설명
배열과 두 개의 숫자 M과 K가 주어지면 배열에서 크기가 K(중첩되지 않음)인 최대 M 하위 배열의 합을 찾아야 합니다. (배열의 순서는 변경되지 않습니다.) K는 하위 배열의 크기이고 M은 하위 배열의 개수입니다. 배열의 크기는 m*k 이상이라고 가정할 수 있습니다. 전체 배열 크기가 k의 배수가 아니면 마지막 배열의 일부를 사용할 수 있습니다.
예
주어진 배열이 {2, 10, 7, 18, 5, 33, 0}인 경우. N =7, M =3 및 K =1이면 부분 집합이 -
이므로 출력은 61이 됩니다.{33, 18, 10}
알고리즘
- 주어진 배열의 'index'부터 'index + K'까지의 모든 요소의 합계를 각 인덱스에 포함하는 추정 배열을 만듭니다. 그리고 합계 배열의 크기는 n+1-k입니다.
- 이제 k 크기의 하위 배열을 포함하면 겹치는 하위 배열을 생성하므로 해당 하위 배열의 요소를 다른 하위 배열에 다시 포함할 수 없습니다. 따라서 포함된 하위 배열의 k 요소를 제외하여 재귀 호출을 수행합니다.
- 하위 배열을 제외하면 해당 하위 배열의 다음 k-1 요소를 다른 하위 배열에서 사용할 수 있으므로 해당 하위 배열의 첫 번째 요소를 제외하여 재귀 호출을 수행합니다.
- 드디어 최대값(포함된 합계, 제외된 합계)을 반환
예
#include <bits/stdc++.h> using namespace std; void calculatePresumArray(int presum[], int arr[], int n, int k) { for (int i = 0; i < k; i++) { presum[0] += arr[i]; } for (int i = 1; i <= n - k; i++) { presum[i] += presum[i-1] + arr[i+k-1] - arr[i- 1]; } } int maxSumMnonOverlappingSubarray(int presum[], int m, int size, int k, int start) { if (m == 0) return 0; if (start > size - 1) return 0; int mx = 0; int includeMax = presum[start] + maxSumMnonOverlappingSubarray(presum, m - 1, size, k, start + k); int excludeMax = maxSumMnonOverlappingSubarray(presum, m, size, k, start + 1); return max(includeMax, excludeMax); } int main() { int arr[] = { 2, 10, 7, 18, 5, 33, 0 }; int n = sizeof(arr)/sizeof(arr[0]); int m = 3, k = 1; int presum[n + 1 - k] = { 0 }; calculatePresumArray(presum, arr, n, k); cout << "Maximum sum = " << maxSumMnonOverlappingSubarray(presum, m, n + 1 - k, k, 0) << endl; return 0; }
위의 프로그램을 컴파일하고 실행할 때. 다음 출력을 생성합니다 -
출력
Maximum sum = 61