Leetcode 560. 和为 K 的子数组

心路历程:

一开始用动态规划求解的,因为很明显这是一个背包问题,但是超内存了

从网上看到需要用数组的前缀和去解决

注意的点:

1、前缀和初始字典为{0:1},含义是第0个前边元素的前缀和为0

2、需要先查找再添加新的前缀和,否则当k=0的时候会重复计算;并且相当于在k=0的时候要自己减去自己,得到空集是没有意义的。

解法:数组前缀和*

python 复制代码
class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 数组前缀和去解决
        pre_sum = 0
        predict = {0: 1} # ?
        res = 0
        for i in range(0, len(nums)):
            pre_sum += nums[i]
            prej = pre_sum - k
            if prej in predict.keys():
                res += predict[prej]
            # 先查找后添加?
            if pre_sum in predict.keys():
                predict[pre_sum] += 1
            else:
                predict[pre_sum] = 1
        return res


                

超内存解法:动态规划+背包

python 复制代码
class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 动态规划+01背包: 超内存
        @cache
        def dp(i, n):  # 以nums[i]为结尾的子数组的和为n的子数组个数
            if i == 0: return int(nums[0] == n)

            if nums[i] == n:  # 可以选也可以不选 i-1
                return 1 + dp(i-1, n-nums[i])  # max(1, 1 + dp(i-1, n-nums[i]))
            else:  # 只能选i-1
                return dp(i-1, n-nums[i])

        count = 0
        for i in range(len(nums)):
            count += dp(i, k)
        return count
相关推荐
军训猫猫头22 分钟前
1.如何对多个控件进行高效的绑定 C#例子 WPF例子
开发语言·算法·c#·.net
success35 分钟前
【爆刷力扣-数组】二分查找 及 衍生题型
算法
Orlando cron1 小时前
数据结构入门:链表
数据结构·算法·链表
牛客企业服务2 小时前
2025年AI面试推荐榜单,数字化招聘转型优选
人工智能·python·算法·面试·职场和发展·金融·求职招聘
糖葫芦君3 小时前
Policy Gradient【强化学习的数学原理】
算法
向阳@向远方4 小时前
第二章 简单程序设计
开发语言·c++·算法
github_czy5 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
许愿与你永世安宁6 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子6 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
满分观察网友z6 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法