LeetCode 0523. 连续的子数组和
Posted Tisfy
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode 0523. 连续的子数组和相关的知识,希望对你有一定的参考价值。
【LetMeFly】523.连续的子数组和
力扣题目链接:https://leetcode.cn/problems/continuous-subarray-sum/
给你一个整数数组 nums
和一个整数 k
,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
- 子数组大小 至少为 2 ,且
- 子数组元素总和为
k
的倍数。
如果存在,返回 true
;否则,返回 false
。
如果存在一个整数 n
,令整数 x
符合 x = n * k
,则称 x
是 k
的一个倍数。0
始终视为 k
的一个倍数。
示例 1:
输入:nums = [23,2,4,6,7], k = 6 输出:true 解释:[2,4] 是一个大小为 2 的子数组,并且和为 6 。
示例 2:
输入:nums = [23,2,6,4,7], k = 6 输出:true 解释:[23, 2, 6, 4, 7] 是大小为 5 的子数组,并且和为 42 。 42 是 6 的倍数,因为 42 = 7 * 6 且 7 是一个整数。
示例 3:
输入:nums = [23,2,6,4,7], k = 13 输出:false
提示:
1 <= nums.length <= 105
0 <= nums[i] <= 109
0 <= sum(nums[i]) <= 231 - 1
1 <= k <= 231 - 1
方法一:哈希 + 前缀和
首先求出原始数组的前缀和 p r e f i x prefix prefix。 p r e f i x [ i ] = ∑ j = 0 j < i n u m s [ j ] prefix[i] = \\sum_j=0^j<i nums[j] prefix[i]=∑j=0j<inums[j](说人话就是: p r e f i x [ i ] prefix[i] prefix[i]是 n u m s nums nums前 i − 1 i-1 i−1个元素的和)
求出前缀和有什么意义呢?如果我们想知道 n u m s nums nums数组从 l l l到 r r r的和 n u m s [ l ] + n u m s [ l + 1 ] + n u m s [ l + 2 ] + ⋯ + n u m s [ r ] nums[l] + nums[l + 1] + nums[l + 2] + \\cdots + nums[r] nums[l]+nums[l+1]+nums[l+2]+⋯+nums[r],那么我们就可以用前缀和数组 p r e f i x [ r ] − p r e f i x [ l − 1 ] prefix[r] - prefix[l - 1] prefix[r]−prefix[l−1]在 O ( 1 ) O(1) O(1)的时间复杂度内求出。
好了,现在我们来看题目,题目要找到是“是否存在 n u m s nums nums数组从 l l l到 r r r的和,这个和是 k k k的整数倍,且 r > l r > l r>l”
那么问题就等价于:“是否存在 l l l和 r r r满足 r > l r>l r>l且 p r e f i x [ r ] − p r e f i x [ l − 1 ] prefix[r] - prefix[l - 1] prefix[r]−prefix[l−1]是 k k k的整数倍”
“ p r e f i x [ r ] − p r e f i x [ l − 1 ] prefix[r] - prefix[l - 1] prefix[r]−prefix[l−1]是 k k k的整数倍”则说明 p r e f i x [ r ] prefix[r] prefix[r]和 p r e f i x [ l − 1 ] prefix[l - 1] prefix[l−1]关于 k k k同余。
等等,同余?
我们直接记录下 p r e f i x prefix prefix中每一个元素关于 k k k的余数不就好了么
遍历一遍 p r e f i x prefix prefix数组,如果 p r e f i x prefix prefix中某个元素关于 k k k的余数在之前出现过,那么就说明存在 p r e f i x [ i ] prefix[i] prefix[i]和 p r e f i x [ j ] prefix[j] prefix[j],其中 j > i j>i j>i且 p r e f i x [ i ] prefix[i] prefix[i]和 p r e f i x [ j ] prefix[j] prefix[j]关于 k k k同余。
但是题目要求是 r > l r>l r>l,也就是说需要满足“ j > i − 1 j>i-1 j>i−1”,即“ j − i ≥ 2 j-i\\geq 2 j−i≥2”
因此,我们将 p r e f i x [ i ] prefix[i] prefix[i]关于 k k k的余数,延迟两个元素再插入哈希表中即可。
- 时间复杂度 O ( n ) O(n) O(n),其中 n n n是数组中元素的个数
- 空间复杂度 O ( n ) O(n) O(n)
AC代码
C++
class Solution
public:
bool checkSubarraySum(vector<int>& nums, int k)
int n = nums.size();
vector<int> prefix(n + 1, 0);
for (int i = 0; i < n; i++)
prefix[i + 1] = prefix[i] + nums[i];
unordered_set<int> se;
for (int i = 2; i <= n; i++)
se.insert(prefix[i - 2] % k);
if (se.count(prefix[i] % k))
return true;
return false;
;
同步发文于CSDN,原创不易,转载请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/127979626
以上是关于LeetCode 0523. 连续的子数组和的主要内容,如果未能解决你的问题,请参考以下文章
LeetCode 523 连续的子数组和[前缀和 哈希表] HERODING的LeetCode之路