525 连续数组
问题描述:
给定一个二进制数组 nums
, 找到含有相同数量的 0
和 1
的最长连续子数组,并返回该子数组的长度。
示例1:
输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量0和1的最长连续子数组。
示例2:
输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。
思路:
【前缀和+哈希表】
最近都是前缀和+哈希表的问题。其实拿到这个问题最初想的是滑窗,毕竟是连续的嘛,所以滑窗是一定能做出来的,但一般滑窗是n的三次方,优化过后也是n²,显然是不可以的,所以这里我们可以使用前缀和的思想。
我们先计算出数组的前缀和,因为这个集合的自由度是2,所以如果进行遍历的话,时间复杂度依然是n²级别的,所以这里我们需要引入哈希表进行操作。
遍历到某个位置,如果这个位置的前缀和与之前某个位置的前缀和是相同的,那么他们之间的和就是0,这样我们就找到一个这样的区间。那么我们只需要保证这个区间最大即可。这样我们就可以在n的复杂度下,将问题解决
但是,实现这个操作需要一个前提,那就是我们需要将所有的0看成-1,这样如果存在相同数量的0和1的话,这个区间的和就是0。有了这样的思路,我们就有了代码的思路,首先应该先将初始值输入,接着遍历即可。
Java代码
/**
* @Description: 力扣525题题解
* @return: 返回结果
* @Author: Mr.Gao
* @Date: 2021/6/3
*/
public int findMaxLength2(int[] nums) {
Map<Integer,Integer> map = new HashMap<>();
map.put(0, -1);
int n = nums.length;
int s = 0;
int re = 0;
for (int i = 0; i < n; i++) {
if(nums[i]==1){
s++;
}else{
s--;
}
if(map.containsKey(s)){
re = Math.max(i-map.get(s),re);
}else{
map.put(s,i);
}
}
return re;
}