题目链接
思路:前缀和+hash表
分析:数组中只有0和1两个数字,那么将数组前缀和的意思就表示到当前位置为止,一共有多少个1,因为只有0和1两个数字,那么当前位置-当前位置1的个数+1就是到当前位置一共有多少个0.再将遍历每个位置,1的个数比0的个数多几个。
简单说一下上面的话:
1.统计当前位置为止,1的个数
2.统计当前位置为止,0的个数
3.统计当前位置为止,1的个数比0的个数多几个(存在tar数组中)
然后遍历tar数组
1.看当前位置是否等于0,如果等于0,那么说明当前位置为止,1和0的个数相等。更新res,并且把0此时的当前为止放在hash表中
2.查看hash表中是否存在当前1和0 个数的差值,如果存在,说明当前为止到之前那个位置,中间1和0的个数是一样的。更新res,此时不用把当前这个差值在hash表中的值更新。
3.否则,直接放到hash表中
代码:
class Solution {
public int findMaxLength(int[] nums) {
int len = nums.length;
//1的个数
int[] onePresum = new int[len];
onePresum[0] = nums[0];
for(int i = 1;i <len;i++){
onePresum[i] = nums[i] + onePresum[i-1];
}
//0的个数
int[] zeroPresum = new int[len];
for(int i = 0;i < len;i++){
zeroPresum[i] = i+1 - onePresum[i];
}
//记录1的个数比0的个数多几个
int[] tag = new int[len];
for(int i = 0 ; i < len;i++){
tag[i] = onePresum[i] - zeroPresum[i];
}
// key:差值 value:出现的下标
HashMap<Integer, Integer> map = new HashMap<>();
int res = 0;
for(int i = 0; i< len;i++){
if(tag[i] == 0){
res = Math.max(res,i+1);
if(!map.containsKey(0)){
map.put(0,i);
}
}else if(map.containsKey(tag[i])){
res = Math.max(res, i - map.get(tag[i]));
}else{
map.put(tag[i],i);
}
}
return res;
}
}