给你一个二元数组 nums ,和一个整数 goal ,请你统计并返回有多少个和为 goal 的 非空 子数组。
子数组 是数组的一段连续部分。
示例:
输入:nums = [1,0,1,0,1], goal = 2
输出:4
解释:
如下面黑体所示,有 4 个满足题目要求的子数组:
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-subarrays-with-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
暴力法
两层for循环,每次统计从第i个数开始的连续子数组的和,与goal比较。
class Solution {
public int numSubarraysWithSum(int[] nums, int goal) {
int sum = 0;
int res = 0;
for (int i = 0; i < nums.length; i++) {
for (int j = i; j < nums.length; j++) {
sum += nums[j];
if (sum == goal) {
res++;
}
}
sum = 0;
}
return res;
}
}
把时间复杂度从n方变成n
思路:每次统计前i个数的和,减去goal,如果map里存在这样的key,说明有连续子数组符合要求。每次都要把目前为止的前i个数的和存到map里,以待后续用。
class Solution {
public int numSubarraysWithSum(int[] nums, int goal) {
Map<Integer,Integer> map = new HashMap<>();
map.put(0,1);
int sum = 0;
int res = 0;
for(int i=0;i<nums.length;i++){
sum += nums[i];
if(map.containsKey(sum-goal)){
res += map.get(sum-goal);
}
map.put(sum,map.getOrDefault(sum,0)+1);
}
return res;
}
}