力扣HOT100——560.和为k的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2

思路：

前缀和+哈希。我们可以先利用一个数组s存储原数组的前缀和，在遍历s的时候利用一个哈希tmp来记录当前元素s[i]出现的次数，同时寻找它们tmp中是否存在是s[i]-k,如果存在，则答案ans加上tmp[s[i]-k]，即s[i]-k此前出现的次数。

class Solution {
public:int subarraySum(vector<int>& nums, int k) {int ans=0;vector<int> s(nums.size()+1);for(int i=0;i<nums.size();i++){s[i+1]=s[i]+nums[i];}unordered_map<int,int> tmp;for(int x:s){ans+=tmp.contains(x-k)?tmp[x-k]:0;tmp[x]++;}return ans;}
};

复杂度分析

• 时间复杂度：O(n)，其中 n 为 nums 的长度。
• 空间复杂度：O(n)。