题目描述

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

示例 1:

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

示例 2:

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

解题思路

把 0 当作 -1，原题即变为求和为 0 的最长连续子数组。

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
/**
 * @param {number[]} nums
 * @return {number}
 */
var findMaxLength = function (nums) {
  let l = 0;
  let sum = 0;

  const map = new Map();
  map.set(sum, -1);

  const n = nums.length;
  for (let i = 0; i < n; i++) {
    nums[i] ? sum++ : sum--;
    if (map.has(sum)) {
      const lastIndex = map.get(sum);
      l = Math.max(l, i - lastIndex);
    } else {
      map.set(sum, i);
    }
  }

  return l;
};