题目来源:https://leetcode-cn.com/problems/binary-subarrays-with-sum/
大致题意:
给定一个由0和1构成的数组和一个goal,判断数组中有多少连续的子序列和为goal。
思路
若子序列和为goal,且子序列起始索引为 i 和 j。那么从索引 0 至 i-1 的所有元素和为sub, 从索引 0 至 j 的所有数为sum。可知它们的关系为:
- sub + goal = sum
于是使用一个值sum保存从第一个元素至今所有数组元素和,然后使用sum减去goal得到差值sub,然后使用哈希表获取sub的次数(即当前位置之前,数组有多个前缀序列的和为sub),那么sub的次数也就代表了goal的次数(也就是从数组首位至今,有多少个尾部为当前元素的子序列的序列和为goal)。然后将本轮次数加到结果中。
需要注意,如果子序列起始位为0,那么sub值也就为0,然后在map中查找0可能无法找到,导致本次结果无法统计。所以需要在遍历前在map中加入一个(0,1)对。
这样的时空复杂度都为O(n)。
升级思路
对于序列和为goal的子序列,它的开头和尾部可能是由0组成,每个0都会让统计次数加1,所以如果能直接统计序列开头和尾部的0的个数,然后滑动窗口,则可以在时间复杂度不变的情况下,达到O(1)的空间复杂度。
但是我自己写了一遍好多bug,看题解用了两个滑动窗口,窗口一存的是以当前遍历元素为结尾,元素和为goal的最长的子序列,起始位为left1;窗口二存的是以当前遍历元素为结尾的元素和刚好小于goal(也就是goal-1)的子序列,起始位为left2。然后left2-left1就是本轮统计结果。其实这个值刚好等于 从left1(left1 < 当前位置)位置开始连续的0元素个数 + 1。
代码:
public int numSubarrayWithSum(int[] nums, int goal) {
int num = 0;
// 方法一
// Map<Integer, Integer> sumMap = new HashMap<Integer, Integer>();
// int tempSum = 0;
// sumMap.put(0, 1); // 初始化,这样保证第一个子序列和为goal的序列也可以被统计上
// for(int i = 0; i < nums.length; i++) {
// tempSum += nums[i]; // 当前总的序列和
// int subNum = tempSum - goal; // 减去goal得差值
// int count = sumMap.getOrDefault(subNum, 0); // 判断在此之前有多少个差值,若无则为0
// num += count; // 统计
// sumMap.put(tempSum, sumMap.getOrDefault(tempSum, 0) + 1); // 将当前序列和放入map
// }
// 方法二
int left1, left2;
int right; // 当前位置
int sum1, sum2;
left1 = left2 = right = sum1 = sum2 = 0;
while (right < nums.length) {
sum1 += nums[right];
while (left1 <= right && sum1 > goal) { // 窗口一
sum1 -= nums[left1++];
}
sum2 += nums[right];
while (left2 <= right && sum2 >= goal) { // 窗口二
sum2 -= nums[left2++];
}
num += left2 - left1;
right++;
}
return num;
}