程序员代码面试指南刷题--未排序数组中累加和为给定值的最长子数组系列问题补2

题目描述
给定一个无序数组arr，其中元素只能是1或0。求arr所有的子数组中0和1个数相等的最长子数组的长度
[要求]
时间复杂度为O(n)，空间复杂度为O(n)
输入描述:

第一行一个整数N，表示数组长度
接下来一行有N个数表示数组中的数

输出描述:

输出一个整数表示答案

示例1

输入

5
1 0 1 0 1

输出

4

解法一：利用map

思路：把0当-1,求相加和为0的最大数组长度

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int len = Integer.parseInt(br.readLine());
        String[] ss = br.readLine().trim().split(" ");
        int[] arr = new int[len];
        for(int i=0;i<arr.length;i++){
            arr[i] = Integer.parseInt(ss[i]);
        }
        int res = getRes(arr);
        System.out.println(res);
    }
    public static int getRes(int[] arr){
        if(arr==null||arr.length==0) return 0;
        HashMap<Integer,Integer> map = new HashMap<>();
        map.put(0,-1);
        int sum = 0;
        int res = 0;
        for(int i=0;i<arr.length;i++){
            sum += arr[i]==0?-1:1;
            if(map.containsKey(sum)){
                res = Math.max(res,i-map.get(sum));
            }else{
                map.put(sum,i);
            }
        }
        return res;
    }
}