给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int n = nums.size();
if(n == 1)
return false;
if(n == 2)
return (nums[0] + nums[1]) % k == 0;
vector<int>dp(n + 1, 0);
vector<bool>modk(k, false);
dp[1] = nums[0] % k;
dp[2] = (nums[0] + nums[1]) % k;
int old = dp[1];
int current = dp[2];
if(current % k == 0)
return true;
for(int i = 3; i <= n; i++){
int temp = (dp[i - 1] + nums[i - 1]) % k;
if(temp == 0)
return true;
dp[i] = temp;
modk[old] = true;
old = current;
current = temp;
if(modk[current])
return true;
}
return false;
}
};
经验:
1、欲速则不达也,想着赶紧锤掉结果提交错误了七八回,这要是罚时我能罚出一个小时来,直接把leetcode正确率刷到49%以下了,长个记性。