ในบทช่วยสอนนี้ เราจะเขียนโปรแกรมที่พบว่าอาร์เรย์ย่อยที่ใหญ่ที่สุดมีผลรวมมากกว่า k
มาดูขั้นตอนการแก้ปัญหากัน
- เริ่มต้นอาร์เรย์
- วนซ้ำบนอาร์เรย์และเก็บผลรวมที่แต่ละดัชนีในเวกเตอร์พร้อมกับดัชนี
- จัดเรียงผลรวมด้านบนตามผลรวมและดัชนี
- เริ่มต้นอาร์เรย์เพื่อจัดเก็บดัชนี
- เขียนลูปที่วนซ้ำจนถึง n.
- อัปเดตค่าด้วยดัชนีต่ำสุดของอาร์เรย์ดัชนีด้านบนและดัชนีอาร์เรย์ผลรวมก่อนหน้า
- เริ่มต้นผลรวมเป็น 0
- เขียนลูปที่วนซ้ำจนถึง n.
- เพิ่มองค์ประกอบปัจจุบันเพื่อรวม
- ถ้าผลรวมมากกว่า k.
- ความยาวสูงสุดของอาร์เรย์ย่อยคือ i + 1
- มิฉะนั้นความยาวสูงสุดของ subarray คือ
- ค้นหาดัชนีจากผลรวมก่อนหน้าโดยใช้การค้นหาแบบไบนารี
- ผลรวมที่น้อยกว่าผลรวม - k - 1 คือดัชนีองค์ประกอบที่เราต้องการ
ตัวอย่าง
มาดูโค้ดกันเลย
#include <bits/stdc++.h>
using namespace std;
bool compare(const pair<int, int>& a, const pair<int, int>& b) {
if (a.first == b.first) {
return a.second < b.second;
}
return a.first < b.first;
}
int findIndex(vector<pair<int, int> >& previousSums, int n, int val) {
int start = 0;
int end = n - 1;
int mid, result = -1;
while (start <= end) {
mid = (start + end) / 2;
if (previousSums[mid].first <= val) {
result = mid;
start = mid + 1;
}else {
end = mid - 1;
}
}
return result;
}
int getLargestSubArray(int arr[], int n, int k) {
int maxLength = 0;
vector<pair<int, int> > previousSums;
int sum = 0, minIndexes[n];
for (int i = 0; i < n; i++) {
sum = sum + arr[i];
previousSums.push_back({ sum, i });
}
sort(previousSums.begin(), previousSums.end(), compare);
minIndexes[0] = previousSums[0].second;
for (int i = 1; i < n; i++) {
minIndexes[i] = min(minIndexes[i - 1], previousSums[i].second);
}
sum = 0;
for (int i = 0; i < n; i++) {
sum = sum + arr[i];
if (sum > k) {
maxLength = i + 1;
}else {
int ind = findIndex(previousSums, n, sum - k - 1);
if (ind != -1 && minIndexes[ind] < i) {
maxLength = max(maxLength, i - minIndexes[ind]);
}
}
}
return maxLength;
}
int main() {
int arr[] = { 5, 3, -3, 2, 4, 7 };
int k = 5, n = 6;
cout << getLargestSubArray(arr, n, k) << endl;
return 0;
} ผลลัพธ์
หากคุณเรียกใช้โค้ดด้านบน คุณจะได้ผลลัพธ์ดังต่อไปนี้
6
บทสรุป
หากคุณมีข้อสงสัยใดๆ ในบทแนะนำ โปรดระบุในส่วนความคิดเห็น
