力扣热门100题之和为k的子数组【中等】

大明子又称小码哥2023-07-25 14:18

题目描述

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。

示例 1：

输入：nums = [1,1,1], k = 2

输出：2

示例 2：

输入：nums = [1,2,3], k = 3

输出：2

提示：

解法1 暴力破解

js 复制代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var subarraySum = function(nums, k) {
    let count=0;
    for(let i=0;i<nums.length;i++){
        let sum=0;
        for(let j=i;j>=0;j--){
            sum+=nums[j];
            if(sum==k){
                count++;
            }
        }
    }
    return count;
};

执行情况

解法2 前缀和

js 复制代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var subarraySum = function(nums, k) {
    let count=0;
    const pre=[]
    let sum=0;
    for(let i=0;i<nums.length;i++){
        sum+=nums[i];
        pre[i]=sum;
    }
    for(let i=0;i<pre.length;i++){
            if(pre[i]==k)count++;
            for(let j=0;j<i;j++){
                if(pre[i]-pre[j]==k){
                    count++;
                }
            }
        
    }
    return count;
};

执行情况

前缀和+哈希

js 复制代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var subarraySum = function(nums, k) {
    let count=0;
    let sum=0;
    const data=new Map()
    data.set(0,1);
    for(let i=0;i<nums.length;i++){
        sum+=nums[i];
        if(data.has(sum-k)){
            count+=data.get(sum-k);
        }
        data.set(sum,data.has(sum)?data.get(sum)+1:1);
    }
    return count;
};

执行情况