- 解题思路
-
暴力解法:先生成一个前缀和数组,再从字串长度为L = 2开始到L = len 对数据数组进行遍历。会超时。O(n^2)
-
利用前缀和值+哈希表。
- 其中有一个同余定理,既当a%x == b%x 时,b-a(假设b>a)为x的整数倍。
- 可以遍历数组,每次计算当前前缀和的值sum,
- sum%x在哈希表中时,需要再进行判断当前位置与哈希表中相等的那个之间的距离是否大于等于2
- 如果sum%x不在哈希表中时,把sum%x存入哈希表中
- 哈希表初始化时要存入(0,-1),下标比长度小1,当元素前缀和余数为0时,准确判断子数组的长度,例如[3,6,1], k =3
- 为什么不要更新哈希表中的下标?(贪心的思维是什么)
- 代码解析
- 暴力解法
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int len = nums.size();
vector<int> prefix_sum(len+1,0);
for(int i = 1; i<len+1; ++i){
prefix_sum[i] = prefix_sum[i-1] + nums[i-1];
}
for(int n = 2;n<len+1;++n){
for(int i = n; i<len+1; ++i){
if(((prefix_sum[i]-prefix_sum[i-n])%k) == 0 ||
prefix_sum[i]-prefix_sum[i-n] == 0) return true;
}
}
return false;
}
};
- 前缀和+哈希表
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int len = nums.size();
unordered_map<int, int> m = {{0,-1}};
int sums = 0,rem = 0;
for(int i = 0;i<len;++i)
{
sums += nums[i];
rem = sums%k;
if(m.count(rem)){
int right = m[rem];
if(i-right>=2) return true;
}else{
m[rem] = i;
}
}
return false;
}
};
- 学习总结
除法转换成求余
利用数学知识解题:同余定理