和为 K 的子数组

/*
 * 思路：
 * 虽然Java Stream不适合处理这种涉及前缀和和哈希表的复杂逻辑，
 * 但我们可以通过转换Stream来实现主要逻辑。
 */
import java.util.HashMap;
import java.util.stream.IntStream;

public class SubarraySumEqualsKStream {
    public int subarraySum(int[] nums, int k) {
        HashMap<Integer, Integer> prefixSumCount = new HashMap<>();
        prefixSumCount.put(0, 1);  // 初始前缀和为0，出现一次
        int[] sum = {0};
        int[] count = {0};

        IntStream.of(nums).forEach(num -> {
            sum[0] += num;
            if (prefixSumCount.containsKey(sum[0] - k)) {
                count[0] += prefixSumCount.get(sum[0] - k);
            }
            prefixSumCount.put(sum[0], prefixSumCount.getOrDefault(sum[0], 0) + 1);
        });

        return count[0];
    }
}