560. 和为 K 的子数组 - 力扣(LeetCode) (leetcode-cn.com)
这题暴力解法超时。。。。(将(0,i)所有的情况列一遍,j=i,j>=0,j--)
优化枚举(0,i)所有子集情况,可以用差来看,(0,i)的和可以拆成(0,j)+(j,i),如果sum(j,i)=k,那统计sum(j,i)=k的次数与统计sum(0,j)出现的次数相同,而sum(0,j)=sum(0,i)-k,这样可以统一到统计从0开始累加和sum(i),出现的次数即可。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int>mp;
int res=0;
int sum=0;
mp[0]=1;
for(int i=0;i<nums.size();i++){
sum+=nums[i];
if(mp.find(sum-k)!=mp.end())
res+=mp[sum-k];
mp[sum]++;
}
return res;
}
};