和相同的二元子数组

/*
 * 题目思路：
 * 使用 Java Stream 的方式解决问题。虽然 Java Stream 处理这种问题可能不如直接使用循环方便，但可以通过流的方式简化代码。
 * 同样利用前缀和的概念，配合 Collectors.groupingBy 和 Collectors.counting 等方法统计满足条件的子数组数量。
 */

import java.util.HashMap;
import java.util.Map;
import java.util.stream.IntStream;

public class SolutionStream {
    public int numSubarraysWithSum(int[] nums, int goal) {
        Map<Integer, Long> prefixSumCount = new HashMap<>();
        prefixSumCount.put(0, 1L);
        int[] prefixSum = {0};
        return (int) IntStream.of(nums)
                .map(num -> prefixSum[0] += num)
                .mapToLong(sum -> {
                    long count = prefixSumCount.getOrDefault(sum - goal, 0L);
                    prefixSumCount.put(sum, prefixSumCount.getOrDefault(sum, 0L) + 1);
                    return count;
                })
                .sum();
    }
}