LeetCode-Python-560. 和为K的子数组

一、题目的内容

力扣里面的560题目：

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :

1. 数组的长度为 [1, 20,000]。
2. 数组中元素的范围是 [-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)  

以上这个写法有两个亮点：

1. 这个写法没有写if，但是等价于有if 语句，
   因为若 pre_sum - k 之前存在，就顺利+x，若之前pre_sum - k不存在，则+0
2. 因为之前的hashtable = {0:1}的定义，使得这个写法，相当于包括了上面实现的A部分，即若前i项之和直接等于k，则hashtable 查找的值第一次是为0的，往后遇到则顺利进入累加