leetcode-525、连续数组

题目

给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。

示例 1:

输入: [0,1]
输出: 2
说明: [0, 1] 是具有相同数量0和1的最长连续子数组。

示例 2:

输入: [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

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

题解

count用来表示0和1之间相差多少，遇到1的话count++，遇到0的话，count–，当count=0的时候，表示从0到当前位置的0和1的数目是相等的，如果两个位置的count的值相等，说明这两个位置之间出现的0和1的数目也是相等的。

class Solution {
    public int findMaxLength(int[] nums) 
    {
        HashMap<Integer,Integer> hashMap = new HashMap();
        int count = 0;
        int max = 0;
        for(int i = 0;i<nums.length;i++)
        {
            if(nums[i] == 1)
            {
                count++;
            }
            else
            {
                count--;
            }
            if(hashMap.get(count) == null)
            {
                hashMap.put(count,i);
            }
            else
            {
                if(i - hashMap.get(count)>max)
                    max = i - hashMap.get(count);
            }
            if(count==0 && max<i+1)
                max = i+1;

        }
        return max;
        
    }
}