560. 和为 K 的子数组

560. 和为 K 的子数组

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

子数组是数组中元素的连续非空序列。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2

示例 2：

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

解题思路

使用前缀和方法。
标准的优化思路，前缀和【从第1个元素累加到第n个元素的和，称为第n个前缀和】，只要出现2个前缀和的差=k，则这两个下标之间的数组即满足要求

进一步优化，第n个元素前缀和为isum_n，则1至n-1如果有x个前缀和为k-isum_n，则第n个元素可以在1至n-1之间形成x个和为k的子数组

以上优化都是考虑前缀和差，因此对于nums只有一个元素的情况要单独进行处理

# 双层循环，暴力解法, 绝对超时
def sub_array_sum1(nums, k):
    if not nums:
        return 0
    count = 0
    for i in range(len(nums)):
        for j in range(i, len(nums)):
            if sum(nums[i:j+1]) == k:
                count += 1
    return count
  
 # 单层循环，将到第n个元素的所有前缀和的计数存入字典
 import collections
def sub_array_sum1(nums, k):
	pre_sums = collections.defaultdict(int)
    pre_sums[0] = 1
    pre_sum = 0
    count = 0

    for i in range(len(nums)):
		pre_sum += nums[i]
         count += pre_sums[pre_sum-k]
         pre_sums[pre_sum] += 1
	return count

在数据结构和算法中，前缀（Prefix）和方法（Method）通常与字符串处理和字符串搜索有关。以下是这两个概念在数据结构和算法中的含义：

前缀（Prefix）

在字符串处理中，前缀指的是字符串的任意子串，它可以是整个字符串，也可以是字符串的前几个字符。例如，对于字符串 “abc”，它的前缀可以是 “a”、“ab” 或 “abc”。

方法（Method）

在算法中，方法通常指的是解决问题的步骤或策略。它可以是递归方法、迭代方法、分治方法等。方法可以用于解决各种问题，包括排序、搜索、图算法等。

前缀和方法的关联

在字符串搜索算法中，前缀和方法常常一起使用。例如，在KMP（Knuth-Morris-Pratt）算法中，前缀和方法的结合被用来提高字符串搜索的效率。KMP算法通过构建一个部分匹配表（也称为前缀表），来避免不必要的字符比较，从而提高搜索效率。

总结

在数据结构和算法中，前缀和方法通常与字符串处理和字符串搜索有关。前缀指的是字符串的任意子串，而方法指的是解决问题的步骤或策略。前缀和方法的结合可以用于提高字符串搜索的效率。