Leetcode - 每日一题（525. 连续数组）

原题链接

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

示例 1:

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

示例 2:

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

思路：该题很显然可以看出是前缀和的思路。我们依次读取数组中的每个元素，并得到其前缀和。当元素为空时，前缀和为0，此时我们规定下标为 -1。

代码

class Solution {
    public int findMaxLength(int[] nums) {
    	//zeroMinusOne:前缀和（数值0减去数值1的个数）
        int zeroMinusOne = 0,answer=0;
        //key:前缀和  value:前缀和对应的下标
        HashMap<Integer,Integer> ZeroMinusOneToIndex= new HashMap<>();
        //规定当元素为空时，前缀和为0，下标为1
        ZeroMinusOneToIndex.put(0,-1);
        for(int i=0;i<nums.length;i++){
            if(nums[i]==1){
                zeroMinusOne-=1;
            }else{
                zeroMinusOne+=1;
            }
            //如果有相同的前缀和，则更新答案
            if(ZeroMinusOneToIndex.containsKey(zeroMinusOne)){
                answer = Math.max(i-ZeroMinusOneToIndex.get(zeroMinusOne),answer);
            //如果有不同的前缀和，就更新hashmap
            } else {
                ZeroMinusOneToIndex.put(zeroMinusOne,i);
            }
        }
        return answer;
    }
}