题目:
给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。
思路:
pre[i]:代表nums[0]到nums[i]的和
同余定理:pre[i]%k - pre[j] %k =(pre[i] - pre[j]) %k
若(pre[i] - pre[j]) %k==0 (即pre[i]%k == pre[j] %k )且 j-i >=2,则满足题意
所以要做判断,仅需用字典记录:前缀和对k的余数 ,以及下标 即可
解答:
法1:超时
class Solution:
def checkSubarraySum(self, nums: List[int], k: int) -> bool:
#求前缀和
n=len(nums)
pre=[0]*n
for i in range(n):
if i ==0:
pre[i]=nums[i]
else:
pre[i]=pre[i-1]+nums[i]
if pre[i]%k==0:
return True
if i>1:
for j in range(i-1):
if (pre[i]-pre[j])%k==0:
return True
return False
法2:优化
class Solution:
def checkSubarraySum(self, nums: List[int], k: int) -> bool:
n=len(nums)
#dic中key:前缀和%k的值,val:当前前缀和对应的数组下标
dic={0:-1}
for i in range(n):
if i ==0:
pre=nums[i]
else:
pre+=nums[i]
key=pre%k
if key not in dic:
dic[key]=i
else:
if i-dic[key]>1:
return True
return False