给定一个包含 非负数 的数组和一个目标 整数 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 位有符号整数范围内。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解:
解法实际上是利用了前缀和,dp[i] 存储的是前面i个元素的和,在边累加的过程中,利用 dp[i-j] = dp[i] - dp[j],即从 i 到 j 的元素和等于从 0 到 i 的和 减去 从 0 到 j 的前缀和(i > j),还有一些细节的处理。
class Solution {
public boolean checkSubarraySum(int[] nums, int k) {
int n = nums.length;
if(n <= 1) return false;
if(k == 1 || k == -1) return true;
int[] sum = new int[n];
sum[0] = nums[0];
if(nums[0] == 0 && nums[1] == 0) return true;
for(int i = 1; i < n; i++) {
if(i != n - 1 && nums[i] == 0 && nums[i+1] == 0) return true;
sum[i] = sum[i-1] + nums[i];
if(k != 0 && sum[i]% k == 0) return true;
for(int j = 0; j < i - 1; j++) {
if(k != 0 && (sum[i] - sum[j]) % k == 0) {
return true;
}
}
}
return false;
}
}