fork download
copy 
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. int goodcount(vector<int>&v){
  4.     unordered_map<int,int>m;
  5.     int sum=0,count=0;m[1]=1;
  6.     for(int i=0;i<v.size();i++){
  7.         sum+=v[i];
  8.         if(m.find(sum-i)!=m.end()){
  9.             count+=m[sum-i];
  10.         }
  11.         m[sum-i]++;
  12.     }
  13.     return count;
  14. }
  15. int main(){
  16.     int n;cin>>n;
  17.     vector<int>v(n);
  18.     for(int i=0;i<n;i++)cin>>v[i];
  19.     cout<<goodcount(v)<<endl;
  20. }
Success #stdin #stdout 0.01s 5280KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
0
https://ideone.com/trA6eG
language:
C++14 (gcc 8.3)
created:
1 day 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!