C++で0と1の数が等しい最大のサブ配列

プログラムを完了するための手順を見てみましょう。

• アレイを初期化します。
• 配列内のすべてのゼロを-1にします。
• 以前のインデックスを保存するために、マップを空のマップにします。
• 合計を0に、最大長を0に、終了インデックスを-1に初期化します。
• nまで繰り返すループを作成します。
  • 現在の要素を合計に追加します。
  • 合計が0に等しい場合。
    • 最大長をi+1で更新します。
    • そしてiのインデックスを終了します。
  • 合計が前の合計マップに存在し、i-previousIndexes[sum]が最大長より大きい場合。
    • 最大長と終了インデックスを更新します。
  • それ以外の場合は、合計を前のインデックスマップに追加します。
• 開始インデックス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

結論

チュートリアルに質問がある場合は、コメントセクションにそのことを記載してください。