1.解析
题目大意,求解连续子数组的和等于k的个数。
2.分析
涉及连续子数组的问题利用累加和是最有效的。比较简单的方法是,直接利用二重循环求求解,第一层循环表示起始的位置,第二层循环从该位置开始逐个相加,然后和k判断即可,时间复杂度为O()。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int n = nums.size(), res = 0;
for (int i = 0; i < n; ++i){
int sum = 0;
for (int j = i; j < n; ++j){
sum += nums[j];
if (sum == k)
res++;
}
}
return res;
}
};
可以对上述算法进行优化,主要利用一个技巧:a + b = c, 则b = c - a; 即如果前面存在子数组的和为a, 如果又存在一个子数组的和为c,则[a, c]子数组的范围内必定存在一个子数组的和为b。利用hashtable存储子数组的和以及对应出现的次数,每次对累加和进行检测,如果当前子数组累计和减去k的值存在hashtable当中,则说明必定存在一个子数组的和等于k。
class Solution {
public:
int subarraySum(vector<int>& nums, int k){
unordered_map<int, int> accumulateSum{{0, 1}};
int sum = 0, res = 0;
for (auto num : nums){
sum += num;
if (accumulateSum.count(sum-k)){
res += accumulateSum[sum-k];
}
accumulateSum[sum]++;
}
return res;
}
};