1、描述
给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。
示例 1:
输入:nums = [23,2,4,6,7], k = 6
输出:true
解释:[2,4] 是一个大小为 2 的子数组,并且和为 6 。
来源:力扣(LeetCode)
链接
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2、关键字
数组,和。连续
3、思路
暴力,
前缀和,
取余 保存一个余数到数组下标的hash,同余定理
作差能被整除,就使用同余定理,只要余数相等,这两个数相减之后就能被整除
4、notes
看到和就去想前缀和
一个参考
5、复杂度
时间:O(n)
空间:min(n,k)取决于哈希表
6、code
// 方法1:前缀和+求余hash表
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
size_t n = nums.size();
vector<int>res(n,0);
//res[0]=nums[0]; // 不需要,在后边循环内判断一下进行初始化前缀和元素
unordered_map<int,int>mp; // 保存余数到下标的hash
// mp[0] = -1; // 这个为了记录当前缀和自己能够整除的时候,第16行做出判断之后,这一行可以省了。
for(int i = 0; i < n;i++){
if(i == 0) // 初始化前缀和
res[i] = nums[i];
else
res[i] = res[i-1]+nums[i];
int tem = res[i] % k; // 临时保存余数
//if(i > 0 && tem ==0) return true;
if(i > 0 && tem ==0) return true; // 直接返回true
if(mp.find(tem) == mp.end()){ // hash表中没有同一个余数,
mp[tem] = i;
}
else if(i - mp[tem] >= 2){
return true;
}
}
return false;
}
};
// 方法2:前缀和
// 两层循环就过不去,而方法1,一个for循环,一个hash寻找,可能更快
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
size_t n = nums.size();
vector<int>res(n,0);
res[0]=nums[0];
for(int i = 1; i< n;i++){
res[i] = res[i-1]+nums[i];
}
for(int i =1; i< n; i++){
if(res[i] % k == 0) return true;
for(int j = 0; j < i - 1;j++){
if((res[i] - res[j]) % k == 0)
return true;
}
}
return false;
}
};
//3:暴力通不过
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
size_t n = nums.size();
if(n < 2) return false;
for(int i = 0; i < n-1; i++){
int res = nums[i];
for(int j = i + 1; j < n; j++){
res += nums[j];
if( res % k == 0){
return true;
}
}
}
return false;
}
};