1. 题目
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
参考题目:LeetCode 1248. 统计「优美子数组」(要复习) 的思路
类似题目:
LeetCode 862. 和至少为 K 的最短子数组(前缀和+deque单调栈)
LeetCode 525. 连续数组(前缀和+哈希)
LeetCode 523. 连续的子数组和(求余 哈希)
LeetCode 974. 和可被 K 整除的子数组(哈希map)
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int, int> m;//和,数组个数
int i, sum = 0, count = 0;
m[0] = 1;//边界,和为0的子数组数量为1
for(i = 0; i < nums.size(); ++i)
{
sum += nums[i];//前缀和
// m[sum]++;不能写在这
count += m[sum-k];//前缀和和为sum-k的有多少个,其后面到i处,和为k
m[sum]++;//更新前缀和为sum的数组个数
}
return count;
}
};
96 ms 21.2 MB