#include<bits/stdc++.h> using namespace std; using LL=long long; #define FOR(i,l,r) for(int i=(l);i<=(r);++i) #define REP(i,n) FOR(i,0,(n)-1) #define ssize(x) int(x.size()) template<class A,class B>auto&operator<<(ostream&o,pair<A,B>p){return o<<'('<<p.first<<", "<<p.second<<')';} template<class T>auto operator<<(ostream&o,T x)->decltype(x.end(),o){o<<'{';int i=0;for(auto e:x)o<<(", ")+2*!i++<<e;return o<<'}';} #ifdef DEBUG #define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<'\n';}(x) #else #define debug(...) {} #endif int main() { cin.tie(0)->sync_with_stdio(0); string s; cin >> s; int n = s.length(); LL ans = 0; map<pair<int,int>,int> m = {{{0,0},1}}; pair<int,int> p = {0,0}; REP(i,n) { if (s.at(i) == 'a') { ++p.first; ++p.second; } else if (s.at(i) == 'b') { --p.first; } else { --p.second; } ans += m[p]; ++m[p]; } REP(i,3) { char zak = 'a' + i; char fir = 'a'; if (fir == zak) ++fir; REP(j,n) { map<int,int> m2; int val = 0; m2[0] = 1; int x = j; while (x < n && s.at(x) != zak) { if (s.at(x) == fir) ++val; else --val; ans += m2[val]; ++m2[val]; ++x; } j = x; } } REP(i,3) { char fir = 'a' + i; REP(j,n) { int x = j; while (x < n && s.at(x) == fir) { ++x; } ans += (LL) (x - j + 1) * (x - j) / 2; j = x; } } cout << ans << '\n'; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | #include<bits/stdc++.h> using namespace std; using LL=long long; #define FOR(i,l,r) for(int i=(l);i<=(r);++i) #define REP(i,n) FOR(i,0,(n)-1) #define ssize(x) int(x.size()) template<class A,class B>auto&operator<<(ostream&o,pair<A,B>p){return o<<'('<<p.first<<", "<<p.second<<')';} template<class T>auto operator<<(ostream&o,T x)->decltype(x.end(),o){o<<'{';int i=0;for(auto e:x)o<<(", ")+2*!i++<<e;return o<<'}';} #ifdef DEBUG #define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<'\n';}(x) #else #define debug(...) {} #endif int main() { cin.tie(0)->sync_with_stdio(0); string s; cin >> s; int n = s.length(); LL ans = 0; map<pair<int,int>,int> m = {{{0,0},1}}; pair<int,int> p = {0,0}; REP(i,n) { if (s.at(i) == 'a') { ++p.first; ++p.second; } else if (s.at(i) == 'b') { --p.first; } else { --p.second; } ans += m[p]; ++m[p]; } REP(i,3) { char zak = 'a' + i; char fir = 'a'; if (fir == zak) ++fir; REP(j,n) { map<int,int> m2; int val = 0; m2[0] = 1; int x = j; while (x < n && s.at(x) != zak) { if (s.at(x) == fir) ++val; else --val; ans += m2[val]; ++m2[val]; ++x; } j = x; } } REP(i,3) { char fir = 'a' + i; REP(j,n) { int x = j; while (x < n && s.at(x) == fir) { ++x; } ans += (LL) (x - j + 1) * (x - j) / 2; j = x; } } cout << ans << '\n'; } |