前缀和满足条件子数组

给你一个二元数组 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;
    }
}