560. 和为K的子数组
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
前台样例
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
题目分析:
今天的题还是需要思考的
1.题目说明了要求子数组连续 我们不能断开它。
2.示例1说明了 情况的不同取决于子数组段位置 而不是子数组本身 这样其实更方便求解。
3.数组极限长度 20000 O(n^2)的代码可能超时。
解体思路:
1.当没有思路的时候首先采用蛮力法思考
初始思路–对于数组的每一个元素均向后遍历(以该元素为头),若满足则break,计数器加一,若大于目标了直接break。
//WA版蛮力法
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int len=nums.size();
int res=0;
for(int i=0;i<len;i++)
{
int sum=0;
for(int j=i;j<len;j++)
{
sum+=nums[j];
if(sum==k) {res++;break;}
if(sum>k) break;
}
}
return res;
}
};
提交 答案错误
34 / 81 个通过测试用例
输入:
[28,54,7,-70,22,65,-6]
100
输出:
0
预期:
1
看到负数就明白 有可能累加和大于了目标值但还可以减下了,所以不对。
同理 如果等于了 也不能break,因为后面还有可能出现满足的。
所以 AC蛮力法如下 时间1612ms居然过了
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int len=nums.size();
int res=0;
for(int i=0;i<len;i++)
{
int sum=0;
for(int j=i;j<len;j++)
{
sum+=nums[j];
if(sum==k) res++;
}
}
return res;
}
};
那么蛮力法的缺点其实很明显了,重复计算的段太多了。
优化应该采用hash表查找
copy官方题解了
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int, int> mp;
mp[0] = 1;
int count = 0, pre = 0;
for (auto& x:nums) {
pre += x;
if (mp.find(pre - k) != mp.end()) count += mp[pre - k];
mp[pre]++;
}
return count;
}
};
用pre来存储序列累加合,譬如pre[i]代表的就是从第0个数到第i个数的累加和
而map其实对应着序列段长度以及其出现的次数。
举个例子
数组为: 3,4,-4,4,7,2,-3,1
k值为: 7
当我们逐位遍历直到1时 此时 pre=14 14-k=7
而前面出现7的地方有 3+4=7 3+4-4+4=7 所以此时应该加2.也就是count += mp[pre - k];
也就是说 如果一个数段能分为3段 a-b-c
我们要让k=7 也就是bc=7 而ac=14 所以ab=14-7=7 只要找到了所以满足的ab段也就对应着所有解了。