思路:
1.记前缀和pre[i]为第i个元素之前所有元素的总和,从左到右遍历数组
2.从左到右遍历数组的过程中,如果pre[i]-pre[j]==k,且0<=j<i,这样则找到了一个总和为k的子数组,因为对于一个i可能同时有多个j使得等式成立,所以我们在遍历到第i个元素时需要知道有多少个j满足pre[j]==pre[i]-k。
3.因为pre[j]的求解是通过前面元素的遍历得来的,因此我们在求解每个pre[i]之后可以将其存入hash表,其中键为pre[i]的值,如果hash表map中已经存在了pre[i]为键的元素,则证明前面也存在某些pre[k](k<i)的值与pre[i]相等,所以我们只需要将map[pre[i]]++即可;如果hash表中不存在,则新插入一条记录(pre[i],1).
代码:
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int n=nums.size();
unordered_map<int,int> mp;
mp[0]=1;
int pre=0;
int ans=0;
for(int i=0;i<n;i++)
{
pre+=nums[i];
if(mp.find(pre-k)!=mp.end())
{
ans+=mp[pre-k];
}
if(mp.find(pre)==mp.end())
mp[pre]=1;
else
mp[pre]=mp[pre]+1;
}
return ans;
}
};