R2-前缀和专题

目录

前缀和+哈希表

双指针

ps:

第一眼过去，这题应该能用双指针解出来，应该也能用前缀和解题。

前缀和+哈希表

适用于 nums[i] 值不固定为 0 和 1 的其他情况

class Solution:
    def numSubarraysWithSum(self, nums: List[int], goal: int) -> int:
        n=len(nums)
        prenum=[0]+list(accumulate(nums))
        dict=defaultdict(int,{0:1})
        ret=0
        for i in range(n):
            r=prenum[i+1]
            l=r-goal
            ret+=dict[l]
            dict[r]+=1
        return ret

 

双指针

 

class Solution:
    def numSubarraysWithSum(self, nums: List[int], goal: int) -> int:
       #就是说[l1,l2]---r,左集合的个数代表当前i下所有可能，i移动就行
       n=len(nums)
       ret=l1=l2=s1=s2=0
       for i in range(n):
           s1+=nums[i]
           s2+=nums[i]
           #左集合左右端点的移动
           while l1<=i and s1>goal:
               s1-=nums[l1]
               l1+=1
           while l2<=i and s2>=goal:
               s2-=nums[l2]
               l2+=1
           ret+=l2-l1
       return ret

ps:

1.python功能使用

2.第一种方法中哈希表定义和使用也很ok啊