给你一个整数数组 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
思路:
利用同余定理可知,前缀和(sum[j] - sum[i] )% k = 0, i < j, -> sum[j] % k = sum[i] % k. 当presum[j]和presum[i]取相同的余数时,nums[i+1] ~ nums[j]的子串之和就是k的倍数。
题目需要满足j-i>=2, 所以哈希表的key用于存放余数,value用于存放最早的余数为key的元素的下标。当再次出现相同的key时,这个元素的下标和哈希表中的key的值之差大于等于2即可。
代码:
class Solution:
def checkSubarraySum(self, nums: List[int], k: int) -> bool:
total = 0
presum = dict() # 前缀和取余为key时,最早元素的下标
presum[0] = -1
for i in range(len(nums)):
total += nums[i]
key = (total % k + k) % k
if key in presum:
if i - presum[key] >= 2:
return True
else:
presum[key] = i
return False
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。