C++ 中的連續陣列


假設我們有一個二進位制陣列,我們必須找到具有相等數量的 0 和 1 的連續子陣列的最大長度。因此,如果輸入是 [0,1,0],則輸出將是 2,因為 [0,1] 或 [1,0] 是具有相等數量的 0 和 1 的最大連續陣列。

要解決這個問題,我們將遵循以下步驟 -

  • ret := 0, n := nums 的大小,sum := 0
  • 建立一個地圖 m,設定 m[0] := - 1
  • 對於 i 介於 0 到 nums 的大小 - 1
    • 當 nums[i] 為 1 時,sum := sum + 1,否則 sum := sum – 1
    • 如果 sum 在 m 中,則 ret := ret 和 i – m[sum] 的最大值,否則 m[sum] := i
  • 返回 ret

讓我們看看以下實現以獲得更好的理解 -

示例

 現場演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int findMaxLength(vector<int>& nums) {
      int ret = 0;
      int n = nums.size();
      int sum = 0;
      map <int, int> m;
      m[0] = -1;
      for(int i = 0; i < nums.size(); i++){
         sum += nums[i] == 1 ? 1: -1;
         if(m.count(sum)){
            ret = max(ret, i - m[sum]);
         }else m[sum] = i;
      }
      return ret;
   }
};
main(){
   vector<int> v = {0,1,0,0,1};
   Solution ob;
   cout << (ob.findMaxLength(v));
}

輸入

[0,1,0,0,1]

輸出

4

更新日期:02-May-2020

936 次瀏覽

開始您的 職業

完成課程後獲得認證

開始學習
廣告
© . All rights reserved.