leetcode 560. Subarray Sum Equals K（和等于K的子数组）

蓝羽飞鸟

于 2022-02-10 17:45:00 发布

阅读量325

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/level_code/article/details/122854270

leetcode 专栏收录该内容

490 篇文章

订阅专栏

这篇博客介绍了如何利用积分数组和哈希表解决寻找和为k的连续子数组个数的问题。通过计算累积和并存储在哈希表中，可以将问题转化为两次查找，从而有效地找到所有满足条件的子数组。这种方法特别处理了数组中可能存在的负数和重复和的情况。

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

在这里插入图片描述
给出一个整数数组nums和整数k, 返回和为k的子数组的个数。
子数组是连续的。

思路：
采用积分数组的思想，就是说sum[i] = nums[0] + nums[1] + … + nums[i]
那么从i 到 k这部分子数组的和就等于sum[k] - sum[i]

子数组的和等于k，就相当于sum[k] - sum[i] = k,
那么sum[k] - k = sum[i]
是不是转换成了Two sum的问题呢

区别是这里nums的元素存在负数，就是说可能存在重复的和，所以要用hashMap, key是sum，value是sum出现的次数。
遍历数组，每个元素处求sum，装入hashMap，map中已存在sum时，value加1。
当hashMap中存在sum-k时，结果次数相加。

注意key=0时，value=1，就是不加任何元素的情况。

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