560. Subarray Sum Equals K-优快云博客

本文链接：https://blog.youkuaiyun.com/m0_37821950/article/details/78551457

本文探讨了一道经典算法题目：寻找数组中所有连续子数组，使得这些子数组的和等于给定整数K的方法。文章提供了一种高效解决方案，通过使用哈希表将问题转化为twoSum问题，显著提高了计算效率。

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

原题：

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:

Input:nums = [1,1,1], k = 2
Output: 2

Note:

The length of the array is in range [1, 20,000].
The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

思考过程：

一开始没什么比较好的思路，暴力破解的话应该会TLE。我的思路是创建一个和给定数组大小相同的数组ints[]，ints[n]表示给定数组的前n个元素之和。遍历给定数组，更新ints数组（遍历到i时，ints数组前i个元素都要加nums[i]），最后居然AC了，当然用的时间比较长，时间复杂度是O( n ^ 2)。

错误示范：

int ret = 0;
    public int mySubarraySum(int[] nums, int k) {//不能排序
        int len = nums.length;
        int[] ints = new int[len];
        for (int i = 0;i < len;i++){
            int x = nums[i];
            for (int j = 0;j <= i;j++){
                ints[j] += x;
                if (ints[j] == k) ret++;
            }
        }
        return ret;
    }

后来看到一种思想，是把nums[]数组前n个元素只和当成一个元素存储到哈希表里。然后这个问题就是two Sum问题了！

AC代码：

 public int subarraySum(int[] nums, int k) {
        int ret = 0;
        HashMap<Integer,Integer> hashMap = new HashMap<>();
        hashMap.put(0,1);//如果一个子数组和为k也可以
        int sum = 0;//存储前n个元素的和
        for (int i = 0;i < nums.length;i++){
            sum += nums[i];
            /*下面两行代码相当于：
            if (hashMap.containsKey(sum - k)) ret += hashMap.get(sum - k);
            if (hashMap.containsKey(sum)) hashMap.put(sum,hashMap.get(sum) ++);
            else hashMap.put(sum,1);
            */
            ret += hashMap.getOrDefault(sum - k,0);//有几个k - sum，ret就加几
            hashMap.put(sum,hashMap.getOrDefault(sum,0) + 1);//如果哈希表存储了sum，那它的值为1，否则就是原来的值加1
        }
        return ret;
    }