和为K的连续子数组

/*
 * 题目：和为K的连续子数组
 * 要求：输入可含正数、负数和零的整数数组 nums，以及整数 k；返回元素和恰好等于 k 的非空连续子数组个数，不是返回子数组本身；空数组返回 0。
 * 示例：nums=[1,-1,0]、k=0 时，合法区间为 [1,-1]、[0]、[1,-1,0]，返回 3。
 */
function countSubarraysWithSum(nums, k) {}
console.log(countSubarraysWithSum([1, 1, 1], 2)); // 预期：2
console.log(countSubarraysWithSum([1, -1, 0], 0)); // 预期：3
console.log(countSubarraysWithSum([0, 0], 0)); // 预期：3
console.log(countSubarraysWithSum([], 0)); // 预期：0

function countSubarraysWithSum(nums, k) {
  const frequencies = new Map([[0, 1]]);
  let prefix = 0;
  let count = 0;
  for (const value of nums) {
    prefix += value;
    count += frequencies.get(prefix - k) ?? 0;
    frequencies.set(prefix, (frequencies.get(prefix) ?? 0) + 1);
  }
  return count;
}

// 三组以上可直接运行的验证。
console.log(countSubarraysWithSum([1, 1, 1], 2)); // 预期：2
console.log(countSubarraysWithSum([1, -1, 0], 0)); // 预期：3
console.log(countSubarraysWithSum([0, 0], 0)); // 预期：3
console.log(countSubarraysWithSum([], 0)); // 预期：0