思路
给定数组 nums 和目标值 k,要找到子数组的数量,使其元素之和等于 k。为了解决这个问题,可以使用一个哈希表来存储之前出现的前缀和,并统计出现次数。遍历数组过程中,不断更新前缀和,并检查之前的前缀和中是否存在 sum[i] - k,如果存在,则说明找到了一个子数组的和为 k,累加相应的计数。
解题方法
初始化一个哈希表,用于存储前缀和及其出现次数。同时初始化前缀和为 0 的出现次数为 1。
初始化变量 sum 为 0,用于记录当前位置的前缀和。
初始化变量 ret 为 0,用于记录子数组的数量。
遍历数组 nums:
更新前缀和 sum += nums[i]。
检查当前位置之前是否存在前缀和为 sum - k 的情况,若存在,则累加相应的计数到 ret 中。
更新哈希表中前缀和为 sum 的出现次数。
返回 ret,即为满足条件的子数组数量
复杂度
时间复杂度: O(n)
空间复杂度: O(n)
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int, int> hash; // 统计前缀和出现的次数
hash[0] = 1;
int sum = 0, ret = 0;
for(auto x : nums)
{
sum += x; // 计算当前位置的前缀和
if(hash.count(sum - k)) ret += hash[sum - k]; // 统计个数
hash[sum]++;
}
return ret;
}
};