解题思路:利用前缀和和哈希表
- 把输入数组中所有的0都替换成-1,那么题目就变成了求包含相同数目的-1和1的最长子数组的长度
- 在一个只包含数字1和-1的数组中,如果子数组-1和1的数目相同的话,那么子数组的所有数字之和就是0
- 所以这个题目就变成了求数字之和为0的最长子数组的长度
遍历每一个元素,计算变形过的前缀和,哈希表中存放每个前缀和与该前缀和第一次出现的下标。当该前缀和第二次出现,意味着这两个前缀和的差是0,也就是说在两个前缀和的下表中存在着相同个数的0和1。
class Solution {
public int findMaxLength(int[] nums) {
//利用前缀和 和哈希表
int n = nums.length;
Map<Integer,Integer> map = new HashMap<>();
int maxLength =0;
// 哈希表中存放前缀和,因为第0项的前缀和是0,所以需要先存入
map.put(0,-1);
// 前缀和变量
int persum=0;
for(int i=0;i<n;i++){
if(nums[i]==1){
persum++;
}else{
persum--;
}
if(map.containsKey(persum)){
int prevIndex = map.get(persum);
// 该前缀和在哈希表中已经存在,该前缀和数值第二次出现,意味着,在prevIndex和i之间存在相同个数的0和1;
maxLength = Math.max(maxLength,i-prevIndex);
}else{
// 哈希表中存放前缀和和,该前缀和第一次出现的下标
map.put(persum,i);
}
}
return maxLength;
}
}