LeetCode
- 题目地址:https://leetcode.com/problems/continuous-subarray-sum/#/description
- 问题分析&解决思路:问给一个非负的数组,是否存在长度不小于2的连续子数组,使得它们的和是k的整数呗。思路如下,维护一个sum,顺着加,每次加完之后取模,如果发现模相同,则说明能整除(比如mod的顺序分别是是1,3,6,4,3那么第二次和第五次都是sum mod k都是3,说明第二加到第五是k的整数倍)。然后由于要长度不小于2,通过维护一个pre即可,先查询,再插入上一次的mod结果
- 代码如下
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;
for (auto j = modk.begin(); j != modk.end(); j++)
cout << i << ":" << *j << endl;
}
return false;
}
};