leetcode560 和为K的子数组

最新推荐文章于 2025-11-27 10:55:11 发布

原创最新推荐文章于 2025-11-27 10:55:11 发布 · 116 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

这篇博客介绍了如何使用哈希表解决寻找整数数组中和为k的连续子数组个数的问题。通过维护前缀和，利用哈希表存储已计算的子数组和及其出现次数，从而高效地找出所有符合条件的子数组。代码中展示了具体的实现方法，包括检查当前子数组和减去k是否在哈希表中存在，以及更新哈希表的过程。

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

思路：采用最长前缀和的方法。从头到尾遍历数组，并保存从最开始到当前位置的子数组和，同时建立一个哈希表，保存每部分的子数组和作为键值，另外一个元素则为子数组和出现的次数，然后将子数组和减去k，查看是否在哈希表里存在这个键值，若存在则将子数组个数增加对应键值的次数，然后再将新的子数组和保存到哈希表中。这个顺序在k为0的时候起到决定性作用。另外有可能出现子数组和本身直接就是k的情况，这时候减去为0，因此最开始需要保存一个特殊元素，键值为0，值为1.

代码：

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer,Integer> hmap = new HashMap<>();
        int sum = 0;
        int ans = 0;
        hmap.put(0,1);
        for(int i = 0;i < nums.length;i++)
        {
            sum += nums[i];
            if(hmap.containsKey(sum - k))
            {
                ans += hmap.get(sum - k);
            }
            if(hmap.containsKey(sum))
            {
                hmap.put(sum,hmap.get(sum)+1);
            }
            else hmap.put(sum,1);
        }
        return ans;
    }
}