#include <bits/stdc++.h> using namespace std; #define rep(i, n) for(int i = 0; i < n; i++) #define pb push_back #define debug(x) cout << #x << " " << x << endl #define all(x) x.begin(),x.end() #define ll long long #define vi vector<int> #define vvi vector<vi> #define vvvi vector<vvi> #define vll vector<ll> #define vvll vector<vll> #define ld long double #define ull unsigned long long #define vull vector<ull> #define fi first #define se second #define PII pair<int,int> #define vPII vector<PII> #define getunique(v) {sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end());} #define int128 __int128 map<PII,int> a,b,c,ab,ac,bc,abc; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cout << setprecision(12) << fixed; // freopen("input.txt", "r", stdin); // freopen("output.txt", "w", stdout); int tt = 1; // cin >> tt; while(tt--) { string s; cin >> s; int n = s.size(); a[{0,0}]++; b[{0,0}]++; c[{0,0}]++; ab[{0,0}]++; ac[{0,0}]++; bc[{0,0}]++; abc[{0,0}]++; int cnta = 0, cntb = 0, cntc = 0; ll ans = 0; rep(i,n){ if(s[i] == 'a'){ cnta++; ans += a[{cntb,cntc}]; } else if(s[i] == 'b'){ cntb++; ans += b[{cnta,cntc}]; } else{ cntc++; ans += c[{cnta,cntb}]; } if(s[i] != 'a'){ ans += bc[{cnta, cntb-cntc}]; } if(s[i] != 'b'){ ans += ac[{cntb, cnta-cntc}]; } if(s[i] != 'c'){ ans += ab[{cntc, cnta-cntb}]; } ans += abc[{cnta-cntb, cnta-cntc}]; a[{cntb, cntc}]++; b[{cnta,cntc}]++; c[{cnta,cntb}]++; ab[{cntc,cnta-cntb}]++; bc[{cnta, cntb-cntc}]++; ac[{cntb, cnta-cntc}]++; abc[{cnta-cntb, cnta-cntc}]++; } cout << ans; } }
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 70 71 72 73 74 75 76 77 78 79 80 | #include <bits/stdc++.h> using namespace std; #define rep(i, n) for(int i = 0; i < n; i++) #define pb push_back #define debug(x) cout << #x << " " << x << endl #define all(x) x.begin(),x.end() #define ll long long #define vi vector<int> #define vvi vector<vi> #define vvvi vector<vvi> #define vll vector<ll> #define vvll vector<vll> #define ld long double #define ull unsigned long long #define vull vector<ull> #define fi first #define se second #define PII pair<int,int> #define vPII vector<PII> #define getunique(v) {sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end());} #define int128 __int128 map<PII,int> a,b,c,ab,ac,bc,abc; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cout << setprecision(12) << fixed; // freopen("input.txt", "r", stdin); // freopen("output.txt", "w", stdout); int tt = 1; // cin >> tt; while(tt--) { string s; cin >> s; int n = s.size(); a[{0,0}]++; b[{0,0}]++; c[{0,0}]++; ab[{0,0}]++; ac[{0,0}]++; bc[{0,0}]++; abc[{0,0}]++; int cnta = 0, cntb = 0, cntc = 0; ll ans = 0; rep(i,n){ if(s[i] == 'a'){ cnta++; ans += a[{cntb,cntc}]; } else if(s[i] == 'b'){ cntb++; ans += b[{cnta,cntc}]; } else{ cntc++; ans += c[{cnta,cntb}]; } if(s[i] != 'a'){ ans += bc[{cnta, cntb-cntc}]; } if(s[i] != 'b'){ ans += ac[{cntb, cnta-cntc}]; } if(s[i] != 'c'){ ans += ab[{cntc, cnta-cntb}]; } ans += abc[{cnta-cntb, cnta-cntc}]; a[{cntb, cntc}]++; b[{cnta,cntc}]++; c[{cnta,cntb}]++; ab[{cntc,cnta-cntb}]++; bc[{cnta, cntb-cntc}]++; ac[{cntb, cnta-cntc}]++; abc[{cnta-cntb, cnta-cntc}]++; } cout << ans; } } |