给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。0 始终视为 k 的一个倍数。
提示:
1 <= nums.length <= 10^5
0 <= nums[i] <= 10^9
0 <= sum(nums[i]) <= 2^31 - 1
1 <= k <= 2^31 - 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题:一、遍历数组
/*方法:遍历数组
朴素的方法是遍历数组nums中所有长度大于等于2的子数组,计算子数组的和,判断是否存在元素和可以被k整除的子数组。如下:复杂度为O(l^2),
*/
class Solution {
public boolean checkSubarraySum(int[] nums, int k) {
int len = nums.length;
int sum = 0;
for(int i = 0;i<len;i++){
sum += nums[i];
}
int sumTmp = sum;
for(int j = len-1; j > 0; j--){
for(int h = 0; h < j; h++){
if(sumTmp%k == 0){
return true;
} else {
sumTmp = sumTmp - nums[h];
}
}
sumTmp = sum - nums[j];
sum = sumTmp;
}
return false;
}
}
最后一个测试用例超出时间限制。
二、前缀+哈希表
/*前缀+哈希表,时间复杂度O(l)
定义:数组下标i对应的前缀为数组nums中索引从0至i的子数组。
数组长度为len,0<=i<len,遍历数组中每一个i,计算i对应的前缀数组元素的和,以及除以k的余数,同时将(余数,下标)键值对维护到哈希表中。如果余数为0,自然返回true,如果余数不为零,则判读hashmap中是否已存在等于该余数的key,不存在,则继续维护键值对,存在,则对比两个下标,判断差是否大于等于2,如果满足,则返回true,否则继续遍历。
*/
import java.util.HashMap;
import java.util.Map;
public class Test01 {
public static void main(String[] args) {
// write your code here
int[] nums= {1,0};
Boolean res = checkSubarraySum(nums,2);
System.out.println(res);
}
public static boolean checkSubarraySum(int[] nums, int k) {
int sum = 0;
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int i = 0; i<nums.length; i++){
sum += nums[i];
if(sum%k==0 && i>=1 ){
return true;
}
if(map.containsKey(sum%k)){
int preIndex = map.get(sum%k);
if(i-preIndex>=2){
return true;
}
} else {
map.put(sum%k,i);
}
}
return false;
}
}