fork download
copy 
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. void solve() {
  5.     int n; cin>>n;
  6.     string s; cin>>s;
  7.  
  8.     unordered_map<int,int> mp;
  9.     mp[1]=1;
  10.  
  11.     int sum=0;
  12.     long long ans=0;
  13.  
  14.     for(int i=0;i<n;i++) {
  15.         sum+=s[i]-'0';
  16.         int x=sum-i;
  17.         ans+=mp[x];
  18.         mp[x]++;
  19.     }
  20.  
  21.     cout<<ans<<'\n';
  22. }
  23.  
  24. int main() {
  25.     int t; cin>>t;
  26.     while(t--) solve();
  27. }
Success #stdin #stdout 0.01s 5284KB
comments (?)
stdin
copy 
3
3
120
5
11011
6
600005
stdout
copy 
3
6
1
https://ideone.com/1MxE7M
23 PART 1
language:
C++14 (gcc 8.3)
created:
9 hours 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!