每日一题：Leetcode560.和为K的子数组（前缀和+哈希）

题目描述：

给你一个整数数组?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

思路：遍历数组使用一个变量presum记录当前节点的前缀和，用另一个指针去遍历该位置前面的数组，找到和为k的子数组，但是这样的话时间复杂度就是o（n^2）

优化：采用哈希表记录前缀和，把找和为k的子数组转化为在哈希表中找和为presum-k的前缀和数组，如果哈希表中存在则将结果ret加上该前缀和所对应的哈希值（即出现的次数），再将该位置的前缀和放入哈希表中。注意特殊情况，如果presum==k则将ret++。

代码实现：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int presum=0,ret=0;
        unordered_map<int,int> hash;
        for(auto e: nums)
        {
            presum+=e;
            if(hash.count(presum-k))
                ret+=hash[presum-k];
            if(presum==k)
                ret++;
            hash[presum]++;
        }
        return ret;
    }
};