LeetCode560. 和为 K 的子数组（前缀和+ HashMap）

力扣

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k/solution/he-wei-kde-zi-shu-zu-by-leetcode-solution/
来源：力扣（LeetCode） 

        

 

解题思路:

本题的解题思路需要用到 前缀和 的思想，还不知道的小伙伴可以点击看一下，是一个B站上的视频，讲解很详细。

                                           0 <= j <= i     pre[i] - pre[j] = k ;

        

    1. 我们可以考虑以 i 结尾的和为 k 的连续子数组个数时只要统计有多少个前缀和为   pre[i] - k的 pre[j]  即可。我们建立哈希表 ，以前缀和和为键 -key，出现次数为对应的值 - value，记录出现的次数，从左往右边更新 mp 边计算答案，那么以 i 结尾的答案 mp[ pre[i] - k ]  即可在 O(1) 时间内得到。最后的答案即为所有下标结尾的和为 k 的子数组个数之和。

        

     2.需要注意的是，从左往右边更新边计算的时候已经保证了 mp[ pre[i] - k ]  里记录的 pre[j] 的下标范围是 0 <= j <= i。同时，由于 pre[i]  的计算只与前一项的答案有关，因此我们可以不用建立 pre 数组，直接用 pre 变量来记录 pre[i−1] 的答案即可

        

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) 
    {
        int len=nums.size();
        unordered_map<int,int> preSum;

        preSum[0]=1;   对于一开始的情况，下标 0 之前没有元素，可以认为前缀和为 0，个数为 1 个
        
        int count=0,sum=0;

        for(int i=0;i<len;i++)
        {
            sum += nums[i];
            if(preSum[sum-k]!=0)
            {
                count += preSum[sum-k];
            }
            preSum[sum]++;
        }
        return count;
    }
};