523. 连续的子数组和
难度:中等
语言:java
题目内容
给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。
解法分析
这是一个比较典型的前缀和的问题,连续数组的和能否被整除,前缀和的方法就是构造PreSum数组,我这边先尝试用了暴力法能否快速解决问题,答案显然是No。
class Solution {
public boolean checkSubarraySum(int[] nums, int k) {
int n = nums.length;
int[] presum = new int[n+1];
presum[0] = 0;
for (int i = 1; i<n+1;i++){
presum[i]=presum[i-1]+nums[i-1];
}
for (int i = 0; i<n+1;i++){
for (int j = 2 ; j<n+1-i;j++){
if ((presum[i+j]-presum[i])%k==0)return true;
}
}
return false;
}
}
这里我的复杂度最高是O(n^3),超时也正常,借鉴一下宫水三叶大神的解答,前缀和的部分是
看到有相同的概念,脑子里第一个冒出来的是哈希表,这样是不是只需要将前缀和的每个数取余后放入哈希表中,后面如果有相同的值,就一定可以解决问题。
class Solution {
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;
}
}