1,题目要求
Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.
Example 1:
Input: nums = [1,1,1], k = 2
Output: 2
Note:
- The length of the array is in range [1, 20,000].
- The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].
给定一个整数数组和一个整数** k ,您需要找到其总和等于 k **的连续子数组的总数。
2,题目思路
对于这道题,题目要求非常简单,给定一个数组以及一个整数k,寻找数组中连续数字之和等于k的序列种类数。
如果我们用暴力的方法,则问题变得非常简单,只需要从每个数字开始,以此加和到结尾,判断期间是否可以等于k。
同时,对于这种暴力法,需要注意的是,数组中的数字是可能是负的,因此,在遍历的过程中,不管等不等于k、有没有序列和等于k,都需要遍历到结尾,因此,这种方法的耗时就非常大。
而另外一种方法,就是对这种加和进行分析和转化:
假设对于一个序列sum[i-j], i < j
,则我们可以有:
sum[0 - j] - sum[0 - i]
= sum[j] - sum[i]
针对这一特点,我们就可以对nums数组进行按顺序的遍历,依次找到index从0到i的sum的出现次数,然后判断之前连续的续联中加和等于sum - k
的序列的个数,就可以得到最终的结果了。
3,代码实现
1,暴力
static auto speedup = [](){
ios::sync_with_stdio(false);
cin.tie(nullptr);
return nullptr;
}();
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int res = 0;
for(int i = 0;i<nums.size();i++){
int tmp = 0;
int j = i;
while(j<nums.size()){
tmp += nums[j++];
if(tmp == k)
res++;
}
}
return res;
}
};
2,累加和转化为差
class Solution {
public:
int subarraySum(vector<int> &nums, int k){
map<int, int> sumMap; //按顺序统计index从0到i的sum的出现次数
int res = 0;
int sum = 0;
sumMap[0] = 1; //加和为0的先增加1,因为一定存在
for(auto &n : nums){
sum += n;
res += sumMap[sum - k];
sumMap[sum]++;
}
return res;
}
};