Submission #20386817

---

Source Code Expand

#include "bits/stdc++.h"
using namespace std;

#define ll long long
#define ld long double
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define sz(v) (int)v.size()
#define precision cout << fixed << setprecision(15);

const int inf = 1e9;
const long long INF = 1e18;
const int mod = 1e9 + 7;
const int bit32 = log2(inf) + 3;
const int bit64 = log2(INF) + 3;

mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
clock_t time_p = clock();

void ktj() {
  time_p = clock() - time_p;
  cerr << "Time elapsed : " << (float)(time_p)/CLOCKS_PER_SEC << "\n";
}

void pre() {}

string to_string(string s) { return '"' + s + '"';}
string to_string(char s) { return string(1, s);}
string to_string(const char* s) { return to_string((string) s);}
string to_string(bool b) { return (b ? "true" : "false");}
 
template <typename A> string to_string(A);
 
template <typename A, typename B>string to_string(pair<A, B> p) {
  return "(" + to_string(p.first) + ", " + to_string(p.second) + ")";
}
 
template <typename A> string to_string(A v) {
  bool f = 1; string r = "{"; 
  for (const auto &x : v) { if (!f) r += ", "; f = 0; r += to_string(x); } 
  return r + "}";
}
 
void debug_out() { cerr << endl; }
 
template <typename Head, typename... Tail> void debug_out(Head H, Tail... T) { 
  cerr << " " << to_string(H); debug_out(T...); 
}
 
#ifdef LOCAL
  #define pr(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)
#else
  #define pr(...) 42
#endif

void solve() {
  string s;
  cin >> s;
  vector<int> cnt(26, 0);
  int n = s.size();
  cnt[s[n - 1] - 'a'] = 1;
  long long ans = 0;
  for (int i = n - 3; i >= 0; i--) {
    if (s[i] == s[i + 1]) {
      long long curr = 0;
      curr += (n - 1) - (i + 1);
      curr -= cnt[s[i] - 'a'];
      for (int j = 0; j < 26; j++) {
        if (j != s[i] - 'a')
          cnt[j] = 0;
      }
      cnt[s[i] - 'a'] += curr;
      ans += curr;
    }
    cnt[s[i + 1] - 'a'] += 1;
  }

  cout << ans << '\n';
}

#define GOOGLE 0
#define MULTIPLE_TC 0

int32_t main() {

  ios_base::sync_with_stdio(false);
  cin.tie(NULL);cout.tie(NULL); 

  pre();

  int t = 1, T;

  if (MULTIPLE_TC)
    cin >> t;

  for (T = 1; T <= t; T++) {
    if (GOOGLE)
      cout << "Case #" << T << ": ";
    solve();
  }

  ktj();
}

Submission Info

Judge Result