fork download
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int cs(vector<int>&arr,int k){
  4. unordered_map<int,int>f;
  5. int cnt=0,sum=0;
  6. f[0]=1;
  7. for(int i=0;i<arr.size();i++){
  8. sum+=arr[i];
  9. if(f.find(sum-k)!=f.end()) cnt+=f[sum-k];
  10. f[sum]++;
  11. }
  12. return cnt;
  13. }
  14. int main(){
  15. vector<int>arr={1,2,3,3,2};
  16. cout<<cs(arr,5);
  17. return 0;
  18. }
  19.  
Success #stdin #stdout 0.01s 5288KB
stdin
Standard input is empty
stdout
2