LeetCode 刷题系列（前缀和题目）之 560 和为 K 的子数组

在河之洲木水

已于 2022-04-05 17:55:36 修改

阅读量228

点赞数 1

分类专栏： LeetCode 文章标签： leetcode

于 2022-04-05 17:53:32 首次发布

本文链接：https://blog.csdn.net/qq_33775774/article/details/123973579

版权

LeetCode 专栏收录该内容

246 篇文章 3 订阅

订阅专栏

这篇博客介绍了一种利用哈希表解决LeetCode上的子数组和等于K的问题。通过维护一个记录子数组和的哈希表，可以高效地计算出和为k的子数组个数。博客提供了具体的代码实现，并解释了思路，强调了利用已有的子数组和减少搜索空间的重要性。

摘要由CSDN通过智能技术生成

题目：

给你一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k

思路：

假设数组中有这样一个子数组，nums[i],nums[i+1], ..., nums[j] , 满足子数组的和，sum_i_j == k 的条件，则有 sum_i_j = sum_j - sum_i （其中， sum_n，代表从nums[0], ..., nums[n] 的子数组的和）即 k = sum_j - sum_i, 变换一下有 sum_i = sum_j - k。定义数组子数组和数组 sums, 长度为 nums.length , sums[j] = nums[0] + ... + nums[j] 则原问题统计并返回该数组中和为 k 的子数组的个数，可以理解为求 nums[0], ..., nums[j] 子数组的和 sum_j 减去 k 后得到的 sum_i ，在 sums[0], ..., sums[j-1] 中出现的次数 count，并且 j 从 0 到 nums.length-1 进行遍历，并求和各 count

class Solution {
    public int subarraySum(int[] nums, int k) {
   int count = 0;

        Map<Integer,Integer> map = new HashMap<>();
        map.put(0,1); // sum_i == 0 , 一定存在，数量为1
        
        int sum_j = 0;
        for(int j=0;j<nums.length;j++) {
            sum_j += nums[j];
            int sum_i = sum_j - k;
            if(map.containsKey(sum_i)) {
                count += map.get(sum_i);
            }
            map.put(sum_j,map.getOrDefault(sum_j,0) + 1);
        }

        return count;
    }
}