fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. int main() {
  4. int n,k; cin>>n>>k;
  5. int a[n];
  6. for(int i=0;i<n;i++) cin>>a[i];
  7. unordered_map<int,int> mp;
  8. mp[0]=1;
  9. int s=0,ans=0;
  10. for(int j=0;j<n;j++){
  11. s+=a[j];
  12. ans+=mp[s-k];
  13. mp[s]++;
  14. }
  15.  
  16. cout<<ans;
  17. }
Success #stdin #stdout 0s 5324KB
stdin
5 5
1 2 3 2 5
stdout
3