【LeetCode】第525题——连续数组（难度：中等）

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contiguous-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

把数组中的0变成-1的话，做累加和，若sum[i] = sum[j]，说明 nums[i+1]+nums[i+2]+…+nums[j] = 0，说明1和-1的数量相等，最终说明此区间内0和1的数量相等

sum可以是个数组，也可以是个Map，用Map会更省事。在Map中，所谓的“累加和相等”的概念会变成“key存在”

代码详解

class Solution {
    public int findMaxLength(int[] nums) {
        HashMap<Integer, Integer> hs = new HashMap<>(); // Map的结构为{累加和, 下标}
        int sum = 0; // 累加和
        int ans = 0; // 最长连续数组的长度
        hs.put(0, -1); // 此行代码的作用会在下一小节“注意点”中详细讲述
        for(int i = 0; i < nums.length; ++i) {
            if(nums[i] == 0) { // 把0看成-1
                sum -= 1;
            } else {
                sum += nums[i];
            }
            if(hs.containsKey(sum)) { // 若累加和存在
                ans = Math.max(ans, i - hs.get(sum)); // 取最长的连续子数组
            } else {
                hs.put(sum, i); // 若累加和不存在，则放入Map中
            }
        }
        return ans;
    }
}

注意点

• 发现有一行代码为put(0, -1)，这是做什么？

  若答案子数组就是从nums[0]开始的，若不放入一个key为0的键值对，则对于[0,1]这种的数组就无法得到正确答案
  至于value为何为-1，因为数组下标是从0开始的，要计算个数需要多减去一个-1（就是+1）