leetcode 560. 和为 K 的子数组

leetcode 560. 和为 K 的子数组

写了一个超时的。时间复杂度$O(n^2)$;空间复杂度$O(1)$

//
var subarraySum = function (nums, k) {
  var ans = 0;
  for (let i = 0; i < nums.length; i++) {
    //遍历nums考虑每个位置作为子数组开头的情况
    var sum = 0;
    for (let j = i; j > nums.length; j++) {
      //遍历可能为子数组结尾的位置，并计算sum，如果等于k计入ans
      sum += nums[j];
      if (sum == k) {
        ans++;
      }
    }
  }
  return ans;
};

看了题解发现可以使用前缀和。

var subarraySum = function (nums, k) {
  var ans = 0;
  var pre = [0];
  sum = 0;
  for (let i = 0; i < nums.length; i++) {
    sum += nums[i];
    pre.push(sum);
  }
  for (let i = 0; i <= nums.length; i++) {
    for (let j = i + 1; j <= nums.length; j++) {
      if (pre[j] - pre[i] == k) {
        ans++;
      }
    }
  }
  return ans;
};

欸，果然没这么简单，这个遍历完了还是那么多复杂度啊。又仔细看了看题解，题解还用了哈希表。
前缀和+哈希表优化

var subarraySum = function (nums, k) {
  var ans = 0;
  const mp = new Map();
  mp.set(0, 1);
  let pre = 0;
  for (const x of nums) {
    pre += x;
    if (mp.has(pre - k)) {
      ans += mp.get(pre - k);
    }
    if (mp.has(pre)) {
      mp.set(pre, mp.get(pre) + 1);
    } else {
      mp.set(pre, 1);
    }
  }
  return ans;
};

本次题目对前缀表和哈希有了更多的理解。学到了JS中的Map和for of。