leetcode每日一题—930.和相同的二元子数组

最新推荐文章于 2024-07-04 16:45:09 发布

原创最新推荐文章于 2024-07-04 16:45:09 发布 · 200 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

Python 同时被 2 个专栏收录

484 篇文章

订阅专栏

leetcode

461 篇文章

订阅专栏

这篇博客介绍了如何利用滑动窗口算法解决寻找数组中和为特定目标值的非空子数组数量的问题。提供了两种不同的实现方法，通过维护窗口内的元素和来动态调整窗口边界，从而统计符合条件的子数组个数。

题目：
给你一个二元数组 nums ，和一个整数 goal ，请你统计并返回有多少个和为 goal 的非空子数组。

子数组是数组的一段连续部分。
在这里插入图片描述
思路：
滑动窗口

解答：
方法一：

class Solution:
    def numSubarraysWithSum(self, nums: List[int], goal: int) -> int:
        #求数组A中 元素和小于等于target的连续子数组 的个数
        def atMostK(A, target):
            if target<0:
                return 0
            res=0
            left,right=0,0
            n=len(A)
            while right<n:
                target -= A[right]
                #窗口滑动
                while target< 0:
                    target += A[left]
                    left += 1
                res += right-left + 1
                #窗口拓展
                right+=1
            return res
        return atMostK(nums, goal) - atMostK(nums, goal - 1)

方法二：

class Solution:
    def numSubarraysWithSum(self, nums: List[int], goal: int) -> int:
        #求数组A中 元素和小于等于target的连续子数组 的个数
        def atMostK(A, target):
            if target<0:
                return 0
            res=0
            left,right,total=0,0,0
            n=len(A)
            while right<n:
                total+=A[right]
                #窗口滑动
                while total>target:
                    total-= A[left]
                    left += 1
                res += right-left + 1
                #窗口拓展
                right+=1
            return res
        return atMostK(nums, goal) - atMostK(nums, goal - 1)