Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

ขนาดสูงสุดของ subarray เช่นที่ subarray ทั้งหมดที่มีขนาดนั้นมีผลรวมน้อยกว่า k ใน C++


ในบทช่วยสอนนี้ เราจะพูดถึงโปรแกรมเพื่อค้นหาขนาดอาร์เรย์ย่อยสูงสุด เพื่อให้อาร์เรย์ย่อยทั้งหมดที่มีขนาดนั้นมีผลรวมน้อยกว่า k

สำหรับสิ่งนี้ เราจะได้รับอาร์เรย์ขนาด N และจำนวนเต็ม k งานของเราคือการหาความยาวของ subarray โดยที่ sub-array ทั้งหมดที่มีความยาวนั้นในผลรวมอาร์เรย์ที่กำหนดจะน้อยกว่าหรือเท่ากับ 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