fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main()
  5. {
  6. int n,k;
  7. cin>>k>>n;
  8. vector<int> arr(n);
  9. unordered_map<int,int> mp;// <sum,freq of sum>
  10. cin>>arr[0];
  11.  
  12. for(int i=1;i<n;i++)
  13. {
  14. cin>>arr[i];
  15.  
  16. }
  17. int x=0;
  18. int count=0;
  19. int sum =0;
  20. mp[0]=1; // very important
  21. for(int j=0;j<n;j++)
  22. {
  23. sum=sum+arr[j];
  24. count+= mp[sum-k];
  25. mp[sum]++;
  26. }
  27. cout<<count;
  28.  
  29. return 0;
  30. }
Success #stdin #stdout 0.01s 5284KB
stdin
Standard input is empty
stdout
Standard output is empty