leetcode 560. 和为 K 的子数组
写了一个超时的。时间复杂度 O ( n 2 ) O(n^2) O(n2);空间复杂度 O ( 1 ) O(1) 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。