fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. string s;
  5. long long a[1000005], d = 0;
  6. map<long long, long long> dem, dem1;
  7.  
  8. int main() {
  9.     ios_base::sync_with_stdio(false);
  10.     cin.tie(NULL); cout.tie(NULL);
  11.  
  12. 	cin >> s;
  13.  
  14.     for(int i = 0; i < s.size(); i++) a[i + 1] = s[i] - 48;
  15.  
  16.     dem1[0] = 1;
  17.  
  18.     for(int i = 1; i <= s.size(); i++){
  19.         dem[a[i]]++;
  20.         long long x = dem[1] - dem[0];
  21.         d = d + dem1[x];
  22.         dem1[x]++;
  23.     }
  24.  
  25.     cout << d;
  26.     return 0;
  27. }
  28.  
Success #stdin #stdout 0.01s 5280KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
Standard output is empty
https://ideone.com/7woLVh
language:
C++14 (gcc 8.3)
created:
10 days 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!