C++ 中具有相同數量 0 和 1 的最大子陣列
讓我們看看完成程式的步驟。
- 初始化陣列。
- 將陣列中的所有零更改為 -1。
- 使用一個空對映來儲存之前的索引。
- 初始化 sum 為 0,maxLength 為 0,endingIndex 為 -1。
- 編寫一個迴圈,迭代到 n。
- 將當前元素新增到 sum 中。
- 如果 sum 等於 0。
- 使用 i + 1 更新 maxLength。
- 並將 endingIndex 更新為 i。
- 如果 sum 存在於 previousSums 對映中,並且 i - previousIndexes[sum] 大於 maxLength。
- 更新 maxLength 和 endingIndex。
- 否則將 sum 新增到 previousIndexes 對映中。
- 列印起始索引 endingIndex - maxLength + 1 和結束索引 endingIndex。
示例
讓我們看看程式碼。
#include <bits/stdc++.h> using namespace std; void findTheSubArray(int arr[], int n) { unordered_map<int, int> previousIndexes; int sum = 0, maxLength = 0, endingIndex = -1; for (int i = 0; i < n; i++) { arr[i] = arr[i] == 0 ? -1 : 1; } for (int i = 0; i < n; i++) { sum += arr[i]; if (sum == 0) { maxLength = i + 1; endingIndex = i; } if (previousIndexes.find(sum) != previousIndexes.end()) { if (maxLength < i - previousIndexes[sum]) { maxLength = i - previousIndexes[sum]; endingIndex = i; } }else { previousIndexes[sum] = i; } } cout << endingIndex - maxLength + 1 << " " << endingIndex << endl; } int main() { int arr[] = { 1, 1, 0, 0, 0, 1, 1, 1, 0 }; findTheSubArray(arr, 9); return 0; }
輸出
如果執行以上程式碼,則會得到以下結果。
1 8
結論
如果您在本教程中有任何疑問,請在評論部分中提出。
廣告