一、题目的内容
力扣里面的560题目:
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
- 数组的长度为 [1, 20,000]。
- 数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
二、解题思路
第1个实现的写法
class Solution:
def subarraySum(self, nums, k):
total = 0
res = 0
count = {}
for num in nums:
# 计算前i项目之和
total += num
# A. 若前i项合,直接等于k,则【最终次数】+1
if total == k:
res += 1
# B. 若前i项之和,减去k,曾经也出现过x次,则【最终次数】+x
if total - k in count:
res += count[total - k]
# C. 要记录下,前面的每一个<前i项之和>,好为后面做查询用
# 这个 count.get(total, 0) 的写法绝了,
count[total] = count.get(total, 0) + 1
return res
以上是第一个实现写法
- 整个计算的思路是,以计算<前i项之和>作为主线,顺带着把存在几个子串之和=k给排查出来
- 在逐渐累计<前i项之和>的过程中,若当前的<前i项之和>的尾巴刚好包含k,则去掉k的<和值>一定曾经也是之前的<前i项之和>(也就是出现过)
- 因为存在负数和正数抵消的情况,所以去掉k后,<剩下的曾经的前i项之和>出现的次数可能出现了多次,所以B步骤要+x(即之前出现过的所有次数都加上)
第2个实现的写法
class Solution2:
def subarraySum(self, nums, k):
hashtable = {0:1}
ret = 0
pre_sum = 0
for num in nums:
pre_sum += num
ret += hashtable.get(pre_sum - k, 0)
hashtable[pre_sum] = hashtable.get(pre_sum, 0) + 1
return ret
- 上面第3行这个hashtable 的写法 hashtable = {0:1},实际上等价于上面第一个实现中的 A步骤
ret += hashtable.get(pre_sum - k, 0)
以上这个写法有两个亮点:
- 这个写法没有写if,但是等价于有if 语句,
因为若 pre_sum - k 之前存在,就顺利+x,若之前pre_sum - k不存在,则+0 - 因为之前的hashtable = {0:1}的定义,使得这个写法,相当于包括了上面实现的A部分,即若前i项之和直接等于k,则hashtable 查找的值第一次是为0的,往后遇到则顺利进入累加