一.题目描述‘
给定一个包含非负数的数组和一个目标整数 k,编写一个函数来判断该数组是否含有连续的子数组,其大小至少为 2,总和为 k 的倍数,即总和为 n*k,其中 n 也是一个整数。
示例 1:
输入: [23,2,4,6,7], k = 6
输出: True
解释: [2,4] 是一个大小为 2 的子数组,并且和为 6。
示例 2:
输入: [23,2,6,4,7], k = 6
输出: True
解释: [23,2,6,4,7]是大小为 5 的子数组,并且和为 42。
说明:
数组的长度不会超过10,000。
你可以认为所有数字总和在 32 位有符号整数范围内。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二.思路
动态规划:
下面是O(n^2)时间复杂度的代码,对于元素nums[i],记录以这个元素结尾的所有连续子数组的和,如果为k的倍数,就返回True,每个和可以从nums[i-1]的和中加nums[i]得到,下面的空间复杂度为O(n^2),还可以优化为O(n)
代码如下:
class Solution:
def checkSubarraySum(self, nums: List[int], k: int) -> bool:
Sum=[[nums[i]] for i in range(len(nums))]
for i in range(1,len(nums)):
for j in Sum[i-1]:
if (nums[i]+j==0 and k==0):
return True
if k!=0 and (nums[i]+j)%k==0:
return True
else:
Sum[i].append(nums[i]+j)
return False
时间复杂度为O(n)的解法: