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