525. 连续数组

题目来源：525. 连续数组

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

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

示例 1:

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

示例 2:

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

 

提示：

• 1 <= nums.length <= 105
• nums[i] 不是 0 就是 1