Leetcode 560. 和为K的子数组
题目
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例:
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
- 数组的长度为 [1, 20,000]。
- 数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
题解
哈希表记录前缀和出现的次数,如果当前前缀和为dp[i],之前存在dp[i]-k的前缀和,则只需要答案加上dp[i]-k出现的次数即可。详细过程见代码
代码
int subarraySum(vector<int>& nums, int k) {
int i=0,j=1,n=nums.size(),ans=0;
if(n == 0) return 0;
unordered_map<int,int> preSum; //记录前缀和出现的次数
preSum[0]++;
vector<int> dp(n,0);
dp[0] = nums[0];
preSum[dp[0]]++;
if(dp[0] == k) ans++;
for(int i=1; i<n; i++){
dp[i] = dp[i-1] + nums[i];
ans += preSum[dp[i]-k];
preSum[dp[i]]++;
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。