1.题目描述
给你一个整数数组 nums 和一个整数 k ,请你统计并返回该数组中和为 k 的连续子数组的个数。
示例1:
输入:nums = [1,1,1], k = 2
输出:2
示例2:
输入:[1,-1,0],k=0
输出:3
题目来源:Leetcode
2.题目分析
2.1从穷举到前缀和的转变
- 乍一看道题目相信我们都可以想出时间复杂度为
O(N^2)
的穷举解法,但是对于较大的数组来说,如此高的时间复杂度会比较耗时,因此我们试图优化穷举解法。
注意在题解过程中会出现示例2中的的状态,比如
[1,-1]
和[1,-1,0]
都满足条件,所以说穷举的N的二次方是不能避免的。
- 试想一下我们常用的类似于动态规划的比较高效的解法就其本质而言是去除冗余计算,也就是避免一些结果的多次计算耗费资源。梳理好了这个我们回到问题,首先什么是子数组呢,按照题目要求,若在数组
nums
中存在nums[i]
到nums[j]
这一段的和是k
,则满足和为k的条件,将其转换为数学表达就是公式(1)中的内容,我们想通过多次使用前面计算的值来优化算法,那么此时我们引入一个概念,令Ni
表示数组nums
中下标为i的元素及其前面的元素的总和,这个值被称为前缀和。通过将公式(1)和(2)结合我们可以推出nums的第i项到第j项的元素和如下面的公式(3)所示
公式(1):num[i]+num[i+1]+…+num[j-1]+num[j] = k;(i<=j)
公式(2):Ni = num[0]+num[1]+…+num[i]
公式(3):k=num[i]+num[i+1]+…+num[j-1]+num[j] = Nj-Ni-1
2.2题目转化——前缀和的应用
此时我们利用上面提到的前缀数组的思想就可以得出,当i到j的子数组满足公式(3)的要求时,这个子数组就是一个和为k
的子数组。也就是说我们在解题时只需要求出每个元素的前缀和,然后就可以在后面利用前缀和的差进行数组个数计算。这样似乎已经优化了解题,但是一计算复杂度发现并没有被优化。
我们再回到题目本身,我们会发现,我们关心的其实并不是前缀和之差是不是等于k,我们求这个只是为了计算有多少个前缀和为k的子数组,换句话说我们关心的是在j
之前满足Ni ==Nj- k
的有多少个。这样我们就很容易想到一种数据结构—