题目:给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。输入:nums = [1,1,1], k = 2 输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
这个问题可以通过Prefix sum来求解。
假设我们令P[i] = A[0] + A[1] + … + A[i] 和 P[j] = A[0] + A[1] + … + A[j],
那么P[j] - P[i] = A[i] + A[i+1] + … + A[j]。
如果P[j] - P[i] == S的话,那么(i,j] 就是我们需要的区间。
所以我们对于每个j,我们只要计算有多少个i使得 P[j] - P[i] == S,这样我们就得到了以P[j]作为右区间并且和为S的区间数。对于A中的每个元素都做同样的处理,最有将所有的结果相加即可。
具体实现上,我们通过hash_map记录P[j]。
初始化的时候要注意一个细节,对于map[0]=1。为什么?
因为当P[j]==S时,P[i]=0并且此时我们的result=1。
C++:
int subarraySum(vector<int>& nums, int k) {
if (nums.size() < 1 || k < -1e7 || k > 1e7)
return 0;
// hash表来存储从0到i的序列和
unordered_map<int, int> map;
map[0] = 1; // sum本身等于k
// 结果,0到索引i的序列数值和
int ans = 0, sum = 0;
for (int i = 0; i < nums.size(); ++i) {
sum += nums[i];
// 如果当前和减去k的值已经在之前存在过,那么加上存在的个数
// 为什么是sum-k? 因为两个序列和相减才得到一个区间和,k就是区间和
if (map.count(sum - k) != 0) {
ans += map[sum - k];
}
// sum的个数+1(unordered_map默认初值时0)
++map[sum];
}
return ans;
}
Java:
public int subarraySum(int[] nums, int k) {
int ans = 0;
int sum = 0;
Map<Integer, Integer> hash = new HashMap<>();
hash.put(0, 1);
for (int i = 0; i < nums.length; ++i) {
sum += nums[i];
ans += hash.getOrDefault(sum - k, 0);
hash.put(sum, hash.getOrDefault(sum, 0) + 1);
}
return ans;
}