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
相关推荐
CoovallyAIHub19 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP20 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo20 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo20 小时前
300:最长递增子序列
算法
CoovallyAIHub1 天前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI2 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v2 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农2 天前
【React用到的一些算法】游标和栈
算法·react.js