最大子数组大小,要求该大小的所有子数组和都小于 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
广告