#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int main() {
// your code goes here
vector<int>nums={1,1,1,-1,1,2,3,4,5};
int k=3;
unordered_map<int,int>mpp;
mpp[0]=1;
int sum=0;
int ans=0;
for(int i=0;i<nums.size();i++){
sum+=nums[i];
if(mpp.find(sum-k)!=mpp.end())ans+=mpp[sum-k];
mpp[sum]++;
}
cout<<ans;
return 0;
}
I2luY2x1ZGUgPGlvc3RyZWFtPgojaW5jbHVkZTxiaXRzL3N0ZGMrKy5oPgp1c2luZyBuYW1lc3BhY2Ugc3RkOwoKaW50IG1haW4oKSB7CgkvLyB5b3VyIGNvZGUgZ29lcyBoZXJlCgl2ZWN0b3I8aW50Pm51bXM9ezEsMSwxLC0xLDEsMiwzLDQsNX07CglpbnQgaz0zOwoJCgl1bm9yZGVyZWRfbWFwPGludCxpbnQ+bXBwOwogICAgICAgIG1wcFswXT0xOwogICAgICAgIGludCBzdW09MDsKICAgICAgICBpbnQgYW5zPTA7CiAgICAgICAgZm9yKGludCBpPTA7aTxudW1zLnNpemUoKTtpKyspewogICAgICAgICAgICBzdW0rPW51bXNbaV07CiAgICAgICAgICAgIGlmKG1wcC5maW5kKHN1bS1rKSE9bXBwLmVuZCgpKWFucys9bXBwW3N1bS1rXTsKICAgICAgICAgICAgbXBwW3N1bV0rKzsKICAgICAgICB9Cgljb3V0PDxhbnM7CglyZXR1cm4gMDsKfQ==