连续数组

题目

题目来源：力扣（LeetCode）

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。

示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

思路分析

1. 问题是找0和1数量相同的连续⼦数组，如果把0看作-1，那么问题转化为和为0的连续⼦数组
   2. ⽤map记录第⼀次出现某前缀和的下标
   3. 设空数组的和为0，下标为-1
   4. 遍历数组，⽤⼀个变量pre记录当前下标的前缀和，⽤res记录当前最⻓⼦数组⻓度
   5. 若元素为0，则pre - 1;若元素为1，则pre + 1
   6. 判断Map中是否已有此前缀和
   7. 若有，则res = Math.max(res, i - map.get(pre))（i为当前下标）
   8. 若⽆，则map.set(pre, i)（i为当前下标）
   9. 最终得到最⻓⼦数组⻓度为res

/**
 * @param {number[]} nums
 * @return {number}
 */
var findMaxLength = function (nums) {
  const n = nums.length
  const map = new Map()
  map.set(0, -1)
  let pre = 0
  let res = 0
  for (let i = 0; i < n; i++) {
    pre += nums[i] == 0 ? -1 : 1
    if (map.has(pre)) {
      res = Math.max(res, i - map.get(pre))
    } else {
      map.set(pre, i)
    }
  }
  return res
};