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

class Solution {
    //这题我们的基本思想是将数组中的0全部都替换成-1，这样没加入一个1和没加入一个-1都会抵消。我们使用这样一种思想，前i个元素之和为counti，我们需要找到一个j<i,并且countj=counti的最早一个下标。表示counti中的多余的1或者-1个数和countj是相同的所以去除掉就是一个-1和1个数一致的子数组。通过打擂台的方式记录下最大的子数组下标
    //我们使用一个Hash表存储已经出现过的counti的最早出现的地方，key为counti，value为最早出现的下标。我们需要一个哨兵节点（0，-1）来表示起初的和为0下标为-1用于后面方便求取长度
    public int findMaxLength(int[] nums) {
        //创建Hash表值为key为counti，value为最早出现的下标位置,加入哨兵节点
        HashMap<Integer,Integer> countSum = new HashMap<>();
        countSum.put(0,-1);
        //count用于记录counti
        int count = 0;
        //maxLength用打擂台的方式记录下来最大的长度
        int maxLength = 0;
        //遍历数组
        for(int i = 0; i < nums.length ; i++){
            //首先求出当前的counti需要注意的是0需要用-1来代替
            count += nums[i] == 0 ? -1 : 1;
            //查看是否存在countj使得0和1的个数相同，如果没有则直接把自己加入，如果有直接使用不需要更新，这样的目的在于值存储最左端的和以求取更大的长度
            if(countSum.containsKey(count)){
                maxLength = Math.max(maxLength,i - countSum.get(count));
            }else{
                countSum.put(count,i);
            }
        }
        return maxLength;
    }
}