剑指 Offer II 011. 0 和 1 个数相同的子数组

概要

老方法，简单清晰的图解

题目

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

思路

可以复用上一题的逻辑。如果把0置为-1，则变成了计算和为0的的最长连续子数组的长度。

而那个map的key和value保存的值的含义要发生变化。

key依旧是sum，但是value是第一次出现当前sum的下标。这个下标用于计算相同数量0和1的子数组的长度。因为如果当前下标（假设为n）的sum，与前面出现过的下标m的sum，值相同，那么我们可以推断出，下标m+1到n之间的全部元素，0和1的个数相同。

举例子说明。数组[1,-1,-1,-1,1,1,1]，我们计算出sum(n)，如图所示：

颜色相同的为sum相同的下标，可以看出下标从1到6，2到5，3到4之间的元素0和1的数量都相同。那么最长的长度就是6了。

解法

代码

public int findMaxLength(int[] nums) {
    // 用一个map记录某个sum出现的次数
    Map<Integer, Integer> sumLengthMap = new HashMap<>();
    sumLengthMap.put(0, -1);
    int maxLength = 0;
    int sum = 0;
    for (int i = 0; i < nums.length; i++) {
        int add = nums[i] == 0 ? -1 : 1;
        sum += add;
        if (sumLengthMap.containsKey(sum)) {
            maxLength = Math.max(i - sumLengthMap.get(sum), maxLength);
        } else {
            sumLengthMap.put(sum, i);
        }
    }
    return maxLength;
}