【前缀和】560. 和为 K 的子数组

给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的连续子数组的个数 。

 

示例 1:

输入:nums = [1,1,1], k = 2
输出:2
示例 2:

输入:nums = [1,2,3], k = 3
输出:2
 


 

思路:前缀和

前缀和定义:定义前缀和数组pre, 使得pre[i] 为 nums[0],  ...  , nums[i]  的所有数的和,  pre[i]  可以从pre[i-1]  得到, 即 pre[i] = pre[i-1] + nums[i]

 

 

复制代码
 1 from collections import defaultdict
 2 class Solution:
 3     def subarraySum(self, nums: List[int], k: int) -> int:
 4         pre = defaultdict(int)   # 记录前缀和出现的次数
 5         pre[0] = 1  # 前缀和为0的出现1次
 6         ans = 0
 7         cur = 0    # 记录前缀和   
 8         for num in nums:
 9             cur += num
10             if cur - k in pre:   # 由于使用的是 defaultdict(int),字典中不存在key时,默认值为0,此if 可以省略。
11                 ans += pre[cur-k]
12             pre[cur] += 1
13         return ans
复制代码

 

posted @   wangpengcufe  阅读(30)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示