560. 和为 K 的子数组

Co丿Hx

于 2023-01-14 18:11:24 发布

阅读量176

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/qq_23301703/article/details/128687656

leetcode 专栏收录该内容

38 篇文章

订阅专栏

给定一个整数数组nums和目标值k，计算数组中所有连续子数组的和等于k的个数。使用前缀和与哈希表的方法，通过maps存储前缀和出现的次数，从而计算出和为k的子数组数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例 1：

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

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

公式

前i项和定义为preSum[i]
前j项和定义为preSum[j]
则preSum[j] - preSum[i] 为j-i项和
设preSum[j] - preSum[i] = k
则只需找到preSum[j] - k出现次数，即可知道n项和为k的次数
注意：maps[0] = 1 表示和为0出现过一次，考虑边界情况【0】 0。

思路

代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        // 前缀和
        int count=0;
        Map<Integer,Integer> maps = new HashMap();
        int preSum = 0;

        // 0 和为0出现一次
        maps.put(0,1);
        for(Integer num:nums) {

            preSum += num;

            // 问题 转化 preSum[j] - preSum[i] = k 
            // k 为j-i求和
         
            count = count + maps.getOrDefault(preSum - k, 0);

            // 维护前缀和
            // 
            maps.put(preSum, maps.getOrDefault(preSum, 0) + 1);
        }

        return count;

    }
}