leetcode 506.和为 K 的子数组

最新推荐文章于 2024-10-18 10:39:20 发布

liangzw_xidian

最新推荐文章于 2024-10-18 10:39:20 发布

阅读量47

点赞数

分类专栏：前缀和哈希表文章标签： leetcode 算法

本文链接：https://blog.csdn.net/Angel_QAQ/article/details/130001806

版权

哈希表同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

前缀和

2 篇文章 0 订阅

订阅专栏

给定一个整数数组nums和一个整数k，使用前缀和和哈希表优化的方法统计数组中和为k的连续子数组的数量。例如，在数组[1,1,1]和k=2的情况下，答案是2。代码中定义了一个Solution类，包含一个subarraySum方法，通过迭代计算前缀和并更新哈希表来实现解决方案。

摘要由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

前缀和 + 哈希表优化

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        map.put(0,1);
        int pre = 0, count = 0;
        for(int i = 0; i < n; i++) {
            pre += nums[i];
            if(map.containsKey(pre - k)) {
                count += map.get(pre - k);
            }
            map.put(pre, map.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}