给定一个包含非负数的数组和一个目标整数 k,编写一个函数来判断该数组是否含有连续的子数组,其大小至少为 2,总和为 k 的倍数,即总和为 n*k,其中 n 也是一个整数。
示例 1:
输入: [23,2,4,6,7], k = 6 输出: True 解释: [2,4] 是一个大小为 2 的子数组,并且和为 6。
示例 2:
输入: [23,2,6,4,7], k = 6 输出: True 解释: [23,2,6,4,7]是大小为 5 的子数组,并且和为 42。
说明:
- 数组的长度不会超过10,000。
- 你可以认为所有数字总和在 32 位有符号整数范围内。
Reivew:
可以使用dp来做,当遍历到 i = x 时,判断前第x-1到2项和是否可以整除,这样代码比较容易理解
但是时间复杂度为 O(nlogn) ---------代码请看Code1
我们又可以使用余数和哈希表,将时间复杂度将为O(n)
假设前s项和与前s-x项和除以k余数相等,且第s到s-x项不都为0,则代表这x项满足连续子数组和,返回true
---------代码请看Code2
Code1
class Solution {
public boolean checkSubarraySum(int[] nums, int k) {
for (int i = 1; i < nums.length; i++)
for (int j = 0; j < i; j++) {
nums[j] += nums[i];
if (k == 0 && nums[j] == 0 || k != 0 && nums[j] % k == 0)
return true;
}
return false;
}
}
Code2
class Solution {
public boolean checkSubarraySum(int[] nums, int k) {
HashMap<Integer,Integer> map = new HashMap<>();
int preSum = 0;
map.put(0,-1);
for(int i = 0; i < nums.length; i++){
preSum += nums[i];
if(k != 0)preSum %= k;
int prev = map.getOrDefault(preSum,-2);
if(prev != -2){
if(i - prev +1 > 2)return true;
}else map.put(preSum,i);
}
return false;
}
}