LeetCode Everyday - 连续数组

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第27天，点击查看活动详情

连续数组

给定一个二进制数组 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 <= nums.length <= 105
• nums[i] 不是 0 就是 1

解题思路:

• 循环字符串
• 数组中只包含 1，0 两个数字，那么创建一个计数器 counter，循环当前数组，碰到 1 的时候，counter += 1，碰到 0 的时候 counter -= 1。
• 把计数器 counter 和 索引 i 分别作为 key，value 保存在哈希表 map 中。
• 碰到计数器 counter 存在于哈希表 map 中的时候，说明哈希表中的 counter 的索引 和当前项 i 中间的子数组中， 1 和 0 的数量肯定是相等的。
• 判断这个字数组和 maxLen 的长度，两者比较取最大值赋值给 maxLen。
• 循环结束，返回 maxLen

我的答案:

/**
 * @param {number[]} nums
 * @return {number}
 */
var findMaxLength = function (nums) {
    const map = new Map();  // 哈希值，key 为 counter， value 为 数字中的索引
    map.set(0, -1); 
    let counter = 0;   // 计数器
    let maxLen = 0; // 最大长度
    for (let i = 0; i < nums.length; i++) {
        if (nums[i] === 1) {
            counter++; // 当前项为 1 的时候，自增
        } else {
            counter--; // 当前项为 0 的时候，自减
        }
        if (map.has(counter)) { // 判断哈希表中是否有含有当前计数器的哈希值
            const preIndex = map.get(counter);
            maxLen = Math.max(maxLen, i - preIndex);    // 比较最大长度
        } else {
            map.set(counter, i); // 哈希表中没有 当前计数器 的哈希值，保存起来
        }
    }
    return maxLen; // 返回最大长度
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )