题意:
给定n个数及k的值,问这个序列中,能找到多少段闭区间,使得闭区间内所有数字求和是k。
思路:
数据比较大(n是2e5,|k|是1e15),处理一段区间的和,我们首先会想到用前后缀和来加加减减,这里我们具体用后缀和处理,通过求出下标 id 的后缀和 last[id](不含id本身),我们可以知道一段闭区间 [L, R]中所有元素的和为 a[L] + last[L] - last[R],我们需要让这个值等于给定的k。
第二步思路就是怎么找这些元素和等于k的区间个数,暴力枚举2e5阶和的复杂度肯定会TLE(枚举就能AC不配当D题)。所以要想怎么用空间换时间,对于每一个区间左端点L,我们通过对式子 a[L] + last[L] - last[R] 移项,发现我们可以将要找的和等于k转化为找满足式子 last[R] = a[L] + last[L] - k 的 last[R]且必须要保证R >= L,那么我们可以将每个last[R]和其出现的位置做一个一对多的映射,即 map <LL, vector<int> > pos; 将vector 结合进map 还有个隐藏的好处就是不用单独处理负值的last[R]了,然后我们通过 lower_bound() 返回当前 last[R] 对应的位置中离L最近且不小于L的下标id,再用vector的size减去id即为个数,每次累加进ans即可。
AC代码: