fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. 	int arr[]= {1, 1, 1};
  6. 	int n=sizeof(arr)/sizeof(arr[0]);
  7. 	int count=0;
  8. 	int currentSum=0;
  9. 	unordered_map<int, int> mp;
  10. 	mp[1]=1;
  11. 	for(int j=0;j<n;j++){
  12. 		currentSum = currentSum + arr[j];
  13. 		int rhs = currentSum - j;
  14. 		if(mp.find(rhs)!=mp.end()){
  15. 			count = count + mp[rhs];
  16. 		}
  17. 		mp[rhs]++;
  18. 	}
  19. 	cout<<count;
  20. 	return 0;
  21. }
  22.  
Success #stdin #stdout 0s 5324KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
6
https://ideone.com/S8ZZHW
language:
C++ (gcc 8.3)
created:
1 hour ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!