剑指Offer专项突破版（11）—— 0 和 1 个数相同的子数组

题目

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

思路

我们可以将0看做-1,1还是看过1，这样一个0和1个数相同的子数组，其和就为0

等于是将题目转化为：求和为0的最长的子数组

可以在扫描数组时累加已经扫描过的数字之和。如果数组中前i个数字之和为x，前j个数字（j>i）之和也为x，那么从第i+1个数字到第j个数字的子数组的数字之和为0，这个和为0的子数组的长度是j-i

可以把数组从第1个数字开始到当前扫描的数字累加之和保存到一个哈希表中

由于我们的目标是求出最长子数组的长度，因此哈希表需要保存某一个前缀和第一次出现的位置

代码

public int findMaxLength(int[] nums) {
    Map<Integer,Integer> map = new HashMap<>();
    map.put(0,-1);
    int sum = 0;
    int res = Integer.MIN_VALUE;
    for (int i = 0;i<nums.length;i++) {
        // 累加前缀和
        sum += nums[i] == 1 ? 1 : -1;
        // 如果已经保存过前缀和为sum的下标，那当前就不是第一次出现的位置，不用保存
        // 而是计算长度，看是不是最长的
        if (map.containsKey(sum)) {
            res = Math.max(res, i - map.get(sum));
        } else {
        // 否则就需要保存
            map.put(sum,i);
        }
    }
    
    return res ==  Integer.MIN_VALUE ? 0 : res;
}