LeetCode热题100-560. 和为 K 的子数组

今天_也很困2026-02-01 16:43

中等

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

子数组是数组中元素的连续非空序列。

示例 1：

复制代码

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

示例 2：

复制代码

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

提示：

1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107

思路1：依旧暴力

java 复制代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n=nums.length,ans=0,sum=0;
        for(int i=0;i<n;i++){
            sum=nums[i];
            for(int j=i+1;j<n;j++){
                if(sum==k) ans++;
                sum+=nums[j];
            }
            if(sum==k) ans++;
        }
       
        return ans;
    }
}

思路2：前缀和

java 复制代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n=nums.length,res=0;
        int[]pre = new int[n+1];//前缀和数组
        for(int i=0;i<n;i++){
            pre[i+1]=nums[i]+pre[i];
        } 
        Map<Integer,Integer> map = new HashMap<>();
        for(int t : pre){
            int s=t-k;
            if(map.containsKey(s)){
                res+=map.get(s);
            }
            map.put(t,map.getOrDefault(t,0)+1);
        }
        return res;
    }
}