这道题虽然在动态规划的分类里,但没有用到动态规划。本质是数学题。
这里用到了取余的知识。从头开始记录sum的值,如果在一个位置,这里的和模k取余的值在前面某个数的相同,则这两个数之间的和是k的倍数。
比较麻烦的是需要考虑特殊情况。比如k是负数时,我们可以直接取绝对值,因为在模的概念下正负是一样的。如果k是0,就不能取余,直接用sum的值记录。还需要注意的是,开头的set中需要先加入0的值,因为有可能前几个数的和就是k的倍数。
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
if(k < 0)k = -k;
if(k == 0){
for (int i = 0; i < nums.size() - 1; ++i)
if(nums[i] == 0 && nums[i + 1] == 0)return true;
return false;
}
set<int> mod;
mod.insert(0);
int sum = 0;
for (int i = 0; i < nums.size(); ++i)
{
sum += nums[i];
if(mod.find(sum % k) != mod.end() && i >= 1)return true;
else mod.insert(sum % k);
}
return false;
}
};