题目描述
给定一个无序数组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;
}
}