525. Contiguous Array

Given a binary array, find the maximum length of a contiguous subarray with equal number of 0 and 1.

Example 1:

Input: [0,1]
Output: 2
Explanation: [0, 1] is the longest contiguous subarray with equal number of 0 and 1.

Example 2:

Input: [0,1,0]
Output: 2
Explanation: [0, 1] (or [1, 0]) is a longest contiguous subarray with equal number of 0 and 1.

Note: The length of the given binary array will not exceed 50,000.

题目：给定一个01串，只含有0和1，求最长的的连续子串，并且子串的0和1的数目相等。

在解决这个题之前，我们先考虑另外一个问题：给定一个序列，其中有正数、负数和0。请你求出和为aim的最长连续子串。

对于这个问题，我们分析一下，假如从序列的第一个数累加到第i个数的和为K，那么我们先在1-i中找到累加和为（K-aim）的最先出现的位置，从这个位置到i的累加和就是aim。也就是说，我们可以找到以第i个位置结尾的累加和为aim的最长子串，把i依次遍历一遍，我们就能取到最长的连续子串了。

01相等子串这个问题也可以用上述的思想来解决，需要做一点小小的改动，就是把01串中的0改成-1，这样一来，题目就变成了求和为0的最长子串。

代码实现：

class Solution {
    public int findMaxLength(int[] nums) {
        HashMap<Integer,Integer> hm = new HashMap<>();
        int max = 0;
        int sum = 0;
        hm.put(0,-1);
        for(int i=0;i<nums.length;i++){
            if(nums[i] == 0)
                nums[i]--;
            sum += nums[i];
            if(hm.containsKey(sum)){
                max = Math.max(max, i-hm.get(sum));
            }else
                hm.put(sum, i);
        }
        return max;
    }
}