523. 连续的子数组和

Difficulty
Medium
Tags
前缀和
URL
https://leetcode-cn.com/problems/continuous-subarray-sum/
Star
给你一个整数数组 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

前缀和

思路
因为需要考虑子数组的长度，而不是计算出现次数，所以字典value存放的是下标。此外，因为需要的是最长子数组，所以只有当当前sum_不在字典中才更新，否则就不去更新。
题解
class Solution:
    def checkSubarraySum(self, nums: List[int], k: int) -> bool:
        hashmap = collections.defaultdict(int)
        hashmap[0] = -1

        sum_ = 0
        for i in range(0, len(nums)):
            sum_ += nums[i]
            temp = sum_ % k
            if temp in hashmap:
                if i - hashmap[temp] >= 2:
                    return True
            else:
                hashmap[sum_%k] = i
        return False