523 连续的子数组和
问题描述:
给你一个整数数组 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 。
示例2:
输入:nums = [23,2,6,4,7], k = 6
输出:true
解释:[23, 2, 6, 4, 7] 是大小为 5 的子数组,并且和为 42 。
42 是 6 的倍数,因为 42 = 7 * 6 且 7 是一个整数。
示例3:
输入:nums = [23,2,6,4,7], k = 13
输出:false
思路:
最先想到的是滑窗,因为题目中有一个描述是连续的子数组,所以最开始想到了滑窗的思路,于是写了个n²的思路,发现爆掉了。无奈只能另辟蹊径。
滑窗Java代码:
/**
* @Description: 力扣523题题解(滑窗爆掉了)
* @return: 返回结果
* @Author: Mr.Gao
* @Date: 2021/6/2
*/
public boolean checkSubarraySum(int[] nums, int k) {
int n = nums.length;
int window = 2;
int sum = nums[0];
for(;window<n;window++){
//表示初始值。
int init = sum+nums[window-1];
//这里将初始值先缓存起来,当滑窗大小变化时,用于下次计算初始值
int temp = init;
//定义滑窗的两个边界
int pro = 0,end = window-1;
if(init%k==0){
return true;
}
end++;
for(;end<n;end++){
init+=nums[end];
init-=nums[pro];
pro++;
if(init%k==0){
return true;
}
}
//将初始值进行调整
sum = temp;
}
return false;
}
- 优化过的滑窗依然超时了。T.T
【前缀和+同余定理】
既然n²过不了,那只能选择比n²更小的时间复杂度了,又要计算某个连续区间的和,能做到这一点的也只有前缀和了,我们计算出数组的前缀和,两个前缀和相减就能得到某个区间里的和,这样再满足这个区间的长度大于2,并且区间的和是k的倍数即可。
这里要判断是k的倍数,我们依然不能遍历所有的区间,因为如果遍历了所有的区间,复杂度同样也是n²级别的,所以这里我们使用同余定理,右边界的值-左边界的值=k的倍数,那么右边界的值余k=左边界的值余k。这样我们只需要枚举右边界即可,将其结果存入哈希表中如果两个数的余数相等,并且相差大于等于2,那么就返回true即可,这样我们就做到了在n的复杂度下的求解。
/**
* @Description: 力扣523题题解
* @return: 返回结果
* @Author: Mr.Gao
* @Date: 2021/6/2
*/
public boolean checkSubarraySum(int[] nums, int k) {
int n = nums.length;
int[] sum = new int[n+1];
for (int i = 1; i <= n; i++) {
sum[i] = sum[i-1]+nums[i-1];
}
Set<Integer> set = new HashSet<>();
for (int i = 2; i <= n; i++) {
set.add(sum[i-2]%k);
if(set.contains(sum[i]%k)) return true;
}
return false;
}