文巾解题 523. 连续的子数组和

1  题目描述

2 解题思路

2.1 前缀和+逐元素遍历

一开始想用前缀和+逐起止坐标来遍历的方法判断有没有符合条件的子数组。（就是先算出a[0], a[0]+a[1], a[0]+a[1]+a[2]+....这些，然后计算sum(nums[i:j])的时候，只需要算前j个的和 减去 前(i-1)个的和就可以了。但是这样会超时。

class Solution:
    def checkSubarraySum(self, nums: List[int], k: int) -> bool:
        tmp=[0]
        for i in nums:
            tmp.append(i+tmp[-1])
#求前缀和


#逐起止坐标遍历
        for i in range(len(nums)-1):
            for j in range(i+1,len(nums)):
                #print(i,j,tmp[j],tmp[i])
                if((tmp[j+1]-tmp[i])%k==0):
                    return(True)
    
        return False

2.2 前缀和+哈希表

换一个思路想，我们记sum(K)是前K个元素的前缀和，那么什么时候sum(nums[i:j])，即sum(j)-sum(i-1)可以被k整除呢？那应该是sum(j)和sum(i-1)被k除后的余数相等。

所以我们可以建立一个哈希表，记录我们sum到下标为几的元素时，这些元素的前缀和被k除后的余数。如果之后的前缀和余数 和之前存在哈希表中的前缀和余数相同，那么我们看一下他们分别sum到那个下标了，如果两个下标的差大于等于2，那么就存在这样的子数组。（比如sum(i)和sum(j)的k余数时一样的，那么i+1~j这些元素组成的子数组满足条件）

有一点需要注意的是，这个哈希表我们得存一个(0,-1),表示我们一个都没有sum的情况下的前缀和。

class Solution:
    def checkSubarraySum(self, nums: List[int], k: int) -> bool:
        dit={0:-1}
        tmp=0
        for i in range(len(nums)):
            tmp=(tmp+nums[i])%k
            if tmp not in dit:
                dit[tmp]=i
            else:
                if(i-dit[tmp]>=2):
                    return(True)
        return False