整体上3个题都是求subarray,都是同一个思想,通过累加,然后判断和目标k值之间的关系,然后查看之前子数组的累加和。
map的存储:560题是存储的当前的累加和与个数
561题是存储的当前累加和的余数与第一次出现这个余数的位置
325题存储的是当前累加和与第一次出现这个和的位置
其实561与325都是求的最长长度,那就一定要存储的是第一次出现满足要求的位置,中间可能还出现这种满足要求的情况,但都不能进行存储
560. Subarray Sum Equals K
求和为k的连续子数组的个数
http://www.cnblogs.com/grandyang/p/6810361.html
如果当前的累加和减去k能找到值,一定是子数组。因为子数组一定是减去前面的值,累加和其实就是看到底减去几个值。
初始化的目的:如果当前位置和为k,那差值就是0,但是存储的值却是sum,不一定存储了0的值。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> m;
m[] = ;
int sum = ,res = ;
for(int i = ;i < nums.size();i++){
sum += nums[i];
res += m[sum - k];
m[sum]++;
}
return res;
}
};
523. Continuous Subarray Sum
思路和560. Subarray Sum Equals K有些类似,因为都是找之前经过的数组的情况,只是560. Subarray Sum Equals K是和。一次循环,加每个数值,然后求余数,如果当前求余数在之前出现过,那当前的数减去之前的子数组的数就可以弥补。
错误解法:
如果一个数在hash_map中没有存储,比如10,使用m[10]不是不返回结果,而是返回0,这个0不代表value值,而是代表没有这个存储。这也是为什么下面这个代码在i = 2时就出错了,i=2时虽然hash_map中没有,但返回0,使得i - 0大于了1
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
unordered_map<int,int> m;
m[] = -;
int sum = ;
for(int i = ;i < nums.size();i++){
sum += nums[i];
int tmp = (k == ) ? sum : sum%k;
if(i - m[tmp] > )
return true;
m[tmp] = i;
}
return false;
}
};
注意1.这个地方m.count(tmp) == 1的时候不要去更新m[tmp] = i,因为这样不能得到最长的长度,去判断有没有至少2个的时候,应该用最远位置去判断。
2. i - (i - 1) >= 1其实就能表示有两个数字,但是这里是需要减去之前的所有,所以必须是i - 1
正确解法:
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
unordered_map<int,int> m;
m[] = -;
int sum = ;
for(int i = ;i < nums.size();i++){
sum += nums[i];
int tmp = (k == ) ? sum : sum%k;
if(m.count(tmp)){
if(i - m[tmp] > )
return true;
}
else
m[tmp] = i;
}
return false;
}
};
325.Maximum Size Subarray Sum Equals k
这个题与523一样unordered_map中只需要保存第一个出现的位置,但保存的是sum,不是target,注意!
注意,你查找的时候是查找sum-k,如果查不到应该分为两种情况,即当前sum在map中有没有值。因为实际上这里是从左向右遍历,只要你找到第一个等于某个sum的,这个值就能成为你需要的最长的子数组。
class Solution {
public:
/**
* @param nums: an array
* @param k: a target value
* @return: the maximum length of a subarray that sums to k
*/
int maxSubArrayLen(vector<int> &nums, int k) {
// Write your code here
unordered_map<int,int> m;
m[] = -;
int res = ;
int sum = ;
for(int i = ;i < nums.size();i++){
sum += nums[i];
if(m.find(sum - k) != m.end()){
res = max(res,i - m[sum-k]);
}
else{
if(m.find(sum) != m.end())
continue;
else
m[sum] = i;
}
}
return res;
}
};