剑指 Offer II 010. 和为 k 的子数组

public int subarraySum(int[] nums, int k) {
    int result = 0, sum = 0;
    Map<Integer, Integer> prefix = new HashMap<>();
    // 前缀和为0的个数 仅作为初始化
    prefix.put(0, 1);

    for (int num : nums) {
        sum += num;
        // 若存在前缀和为sum-k则为解
        result += prefix.getOrDefault(sum - k, 0);
        // 合并前缀和为sum的子数组数量
        prefix.merge(sum, 1, Integer::sum);
    }

    return result;
}