525-连续数组

525. 连续数组

跟523类似，只不过将0的数量减去1的数量的值做前缀和，再用哈希表存储下标。

int findMaxLength(vector<int>& nums) {
  unordered_map<int, int> m;
  int dis = 0, ans = 0, pre = 0;
  m[0] = -1;
  for (int i = 0; i < nums.size(); ++ i) {
    pre += nums[i] == 0 ? 1 : -1;
    if(m.count(pre)) {
      ans = max(ans, i - m[pre]);
    }else m[pre] = i;
  }
  return ans;
}