Leetcode 560. 和为 K 的子数组 JavaScript (Day 5)

POLITE32026-01-03 9:10

javascript 复制代码

var subarraySum = function(nums, k) {
    let count = 0,sum=0;
    let idx=new Map();
    for(let x of nums){
        idx.set(sum,(idx.get(sum)??0)+1);
        sum+=x;
        let j=sum-k;
        count+=idx.get(j)??0;
    }
    
    return count;

}

核心思想:前缀和+两数之和

前缀和
s[0]=0, s[i]=nums[0]+nums[1]+⋯+nums[i−1]
注意 s 是一个长为 n+1 的数组，第一项一定是 0。最后一项的下标是 n。
设 i<j，如果 nums[i] 到 nums[j−1] 的元素和等于 k，用前缀和表示，就是
s[j]−s[i]=k(两数之和等于k)

将所有的前缀和放入哈希表中,边放入边计算,就变成和两数之和一样的思想了

算法注意点:注意当我们拿到右边一个数时,需要先找左边有没有符合要求的数,再把这个数放入哈希表(给出的代码在下一次循环的开始放入哈希表,这样可以避免单独设置(0,1)),一定不能先把这个数放入哈希表后,在来找左边有没有符合要求的数