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

结论

如果您在本教程中有任何疑问，请在评论部分中提出。

更新于: 2021年4月9日

145 次查看

开启你的 职业生涯

通过完成课程获得认证

开始
广告