本题给出一个整数数组,里面的数都是非负整数,另外给出一个整数k,看该数组中是否有连续的子数组和为k或k的倍数。
本题有两个需要注意的地方,一是当k为0的情况,二是子数组长度必须大于1.
思路为用一个集合unsorderd_set来解决。
当k为0时,数组中必须有两个连续的0,unsorderd_set中存放前缀数组和sum.当k不为0时,存放前缀数组和与k的余数,当余数相等时,说明此时的子数组和为k或k的倍数。
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int n = nums.size(), sum = 0, pre = 0;
unordered_set<int> modk;
for (int i = 0; i < n; ++i) {
sum += nums[i];
int mod = k == 0 ? sum : sum % k;
if (modk.count(mod)) return true;
modk.insert(pre);
pre = mod;
}
return false;
}
};