最大子陣列大小,要求該大小的所有子陣列和都小於 C++ 中的 k
在本教程中,我們將討論一個程式,用於找到最大子陣列大小,要求該大小的所有子陣列和都小於 k。
為此,我們將提供一個大小為 N 的陣列和一個整數 k。我們的任務是找到子陣列的長度,要求給定陣列中該長度的所有子陣列的總和小於或等於 k。
示例
#include<bits/stdc++.h> using namespace std; //finding maximum length subarray int bsearch(int prefixsum[], int n, int k) { int ans = -1; //performing binary search int left = 1, right = n; while (left <= right) { int mid = (left + right) / 2; int i; for (i = mid; i <= n; i++) { if (prefixsum[i] - prefixsum[i - mid] > k) break; } if (i == n + 1) { left = mid + 1; ans = mid; } else right = mid - 1; } return ans; } //returning maximum subarray size int maxSize(int arr[], int n, int k) { int prefixsum[n + 1]; memset(prefixsum, 0, sizeof(prefixsum)); for (int i = 0; i < n; i++) prefixsum[i + 1] = prefixsum[i] + arr[i]; return bsearch(prefixsum, n, k); } int main() { int arr[] = {1, 2, 10, 4}; int n = sizeof(arr) / sizeof(arr[0]); int k = 14; cout << maxSize(arr, n, k) << endl; return 0; }
輸出
2
廣告