滑动窗口+前缀和-8--LC930.和相同的二元子数组

zeronose

已于 2022-05-10 12:19:28 修改

阅读量423

点赞数

分类专栏： leetcode 文章标签： leetcode 滑动窗口前缀和

于 2022-05-10 10:53:35 首次发布

本文链接：https://blog.youkuaiyun.com/zeronose/article/details/124683343

版权

leetcode 专栏收录该内容

75 篇文章

订阅专栏

这篇博客介绍了如何利用前缀和和动态规划的方法来解决寻找数组中和为目标值的子数组个数的问题。具体实现包括两种策略：一种是直接利用前缀和进行查找，另一种是针对只包含0和1的数组，通过动态规划计算满足条件的子数组个数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution(object):
    def numSubarraysWithSum(self, nums, goal):
        """
        :type nums: List[int]
        :type goal: int
        :rtype: int
        """
        # 1.前缀和
        presum = 0
        adict = {0:1}
        count = 0
        for i in range(len(nums)):
            presum += nums[i]
            if presum - goal in adict:
                count += adict[presum-goal]
            if presum not in adict:
                adict[presum] = 1
            else:
                adict[presum] += 1
        return count

class Solution(object):
    def numSubarraysWithSum(self, nums, goal):
        """
        :type nums: List[int]
        :type goal: int
        :rtype: int
        """
		# 数组中只有0和1，和就是1的个数
        def leGoal(nums, goal):
            if goal < 0:
                return 0
            start = 0
            one_count = 0
            count = 0
            for end in range(len(nums)):
                if nums[end] == 1:
                    one_count += 1
                while one_count > goal:
                    if nums[start] == 1:
                        one_count -= 1
                    start += 1
                count += end-start+1
            return count
        return leGoal(nums, goal)-leGoal(nums, goal-1)