525. 连续数组

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

我最初的思路为计算出前缀和，当前缀和*2=子数组长度时，即说明子数组中0与1个数相同。但问题在于得到前缀和数组，去逐个遍历计算，时间复杂度仍为O(n**2) ，继续优化的关键为怎么用更多的空间去存储重复的数组，从而减少时间复杂度。

为了能满足优化条件，有个很好的方法是，将0记为-1，当0与1个数相同时，前缀和即为0，同时当两个前缀和相等时，他们的差即为0，则两前缀和下标之间的子数组符合要求。

可以用map去存储各个前缀和以及下标，当两个数前缀和相同，他们下标之间的子数组符合要求了。

class Solution {
    public int findMaxLength(int[] nums) {
        int len=nums.length;
        int maxLength=0;
        int count=0;
        HashMap<Integer,Integer> map=new HashMap<>();
        //当有前缀和为0时，其下标减去-1即得到数组长度。
        map.put(0,-1);
        for(int i=0;i<len;i++){
            if(nums[i]==1){
                count+=1;
            }else{
                count-=1;
            }
            
            if(map.containsKey(count)){
                int pre=map.get(count);
                maxLength=Math.max(maxLength,i-pre);

            }else{
                map.put(count,i);
            }
        }
        return maxLength;
    }
}