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算法通过构建一个部分匹配表(也称为前缀表),来避免不必要的字符比较,从而提高搜索效率。
总结
在数据结构和算法中,前缀和方法通常与字符串处理和字符串搜索有关。前缀指的是字符串的任意子串,而方法指的是解决问题的步骤或策略。前缀和方法的结合可以用于提高字符串搜索的效率。