fork download
copy 
  1. from collections import Counter
  2. def count(s):
  3.     dic = Counter()
  4.     diff = (0,0)
  5.     dic[diff] += 1
  6.     result = 0
  7.     for c in s:
  8.         if c == 'A':
  9.             diff = (diff[0]-1,diff[1]-1)
  10.         elif c == 'B':
  11.             diff = (diff[0]+1,diff[1])
  12.         else:
  13.             diff = (diff[0],diff[1]+1)
  14.         result += dic[diff]
  15.         dic[diff] += 1
  16.     return result
  17.  
  18.  
  19. print(count('CBACBACBA'))
Success #stdin #stdout 0.02s 9024KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
12
https://ideone.com/MOIMqj
language:
Python 3 (python  3.9.5)
created:
3 years 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!