这个月的题都是hash优化
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的最长连续子数组。
提示:
1 <= nums.length <= 105
nums[i] 不是 0 就是 1
思路:
分析成立的条件:命题等价为:找出和的2倍为数组长度的子数组
分析前缀和表达式:
s
u
m
[
j
]
−
s
u
m
[
i
−
1
]
=
(
j
−
i
+
1
)
/
2
sum[j]-sum[i-1]=(j-i+1)/2
sum[j]−sum[i−1]=(j−i+1)/2
我们采用hash优化的思想就是先存i,的表达式,再由j的表达式是否存在于表中,达到等式成立的效果又降低了时间复杂度
2 s u m [ j ] − j = 2 s u m [ i − 1 ] − i + 1 ; 2sum[j]-j=2sum[i-1]-i+1; 2sum[j]−j=2sum[i−1]−i+1;
注意:
(存放i的时候注意,需要判断是否存在它,存在无需覆盖存储)
class Solution {
public int findMaxLength(int[] nums) {
int len=nums.length;
int []sum=new int [len+1];
sum[1]=nums[0];
for(int i=2;i<=len;i++){
sum[i]=sum[i-1]+nums[i-1];
}
HashMap<Integer,Integer> map=new HashMap<Integer,Integer>();
map.put(2*sum[1-1]-1+1,1);
int max=0;
for(int i=2;i<=len;i++){
if(map.containsKey(2*sum[i]-i)){
//得出下标记录,维护max
if(!map.containsKey(2*sum[i-1]-i+1)){
map.put(2 * sum[i-1] - i + 1, i);
}
int j=map.get(2*sum[i]-i);//较小下标
max=Math.max(max,i-j+1);
}
else{
int temp=2*sum[i-1]-i+1;
if(!map.containsKey(2*sum[i-1]-i+1)) {
map.put(2 * sum[i-1] - i + 1, i);
}
}
}
return max;
}
}