题目意思就是求出有多少个子数组的和为K。
题目中给出了样例,就不再多做描述了。
题目其实也能比较容易的想到,因为前缀和求子区段的总和是我们很常见的办法,但是如果一段一段的去看的话,就是说假如已经求出了前缀和数组s[n],然后遍历s[n] - s[n-1] , s[n] - s[n-2]......s[n] - s[1] ... s[2] - s[1] ,s[1] - s[0] , 很明显我们的时间复杂度来到了o(n^2),所以这样的做法肯定是不行的。
思路:
1.因为本题只需要求出子段的和是不是等于k,而不是求出所有子段的可能性,所以我们就要从此下手来完成我们的程序。
2.我们知道k - s[i] 一定是一个固定的值,所以我们只要求出有多少个数等于k -s[i]就可以完成本题目了,因为是前缀和,所以我们就得从最后一个开始,找出前面有多少个s[j]等于k-s[i]。
3.所以我们只需要将我们求出的前缀和的所有值对应的个数标记出来就行了,这里我们要用的STL 用一个map来标记。
4.现在我们完成了每一个数值的标记,所以我们从最后一个数值开始,往前搜索就行,不要忘记每一个数值搜索过之后都要减掉,还有如果k-s[i]=0的时候需要额外加上一个就行了。
#include<iostream>
#include<algorithm>
#include<unordered_map>
const int N = 2e5+5;
typedef long long LL;
LL a[N];
using namespace std;
unordered_map<LL,LL> m;
int main()
{
int n ;
LL k;
cin >>n >> k;
for(int i = 1 ; i <=n ; i ++)
{
scanf("%lld",&a[i]);
a[i] += a[i-1];
m[a[i]] ++;
}
LL ans = 0;
for(int i = n ; i >= 1 ; i --)
{
m[a[i]]--;
if(a[i] == k)
ans ++;
LL x = k - a[i];
ans += m[-x];
}
cout << ans << endl;
return 0;
}