fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. int n;
  6. cin >> n;
  7.  
  8. vector<long long> a(n);
  9.  
  10. for (int i = 0; i < n; i++) {
  11. cin >> a[i];
  12. }
  13. long long k;
  14. cin>>k;
  15.  
  16. vector<long long> p(n);
  17. p[0]=a[0];
  18. for(int i=1;i<n;i++){
  19. p[i]=p[i-1]+a[i];
  20. }
  21.  
  22. int count =0;
  23.  
  24. unordered_map<long long , long long> mp;
  25. for(int j=0;j<n;j++){
  26. long long r = p[j]-k;
  27.  
  28. count+= mp[r];
  29.  
  30. mp[p[j]]++;
  31.  
  32. }
  33. cout<<count<<endl;
  34.  
  35.  
  36. return 0;
  37. }
Success #stdin #stdout 0s 5320KB
stdin
10
2 2 3 5 2 2 3 2 2 1
8
stdout
2