Cod sursa(job #2798847)

Utilizator bogdanvladmihaiBogdan Vlad-Mihai bogdanvladmihai Data 11 noiembrie 2021 23:37:54
Problema PScPld Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.91 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in("pscpld.in");
ofstream out("pscpld.out");

using ll = long long;

vector<int> Manacher(string &s, int add) {
  vector<int> pal((int)s.size());
  int l = -1, r = -1;
  for (int i = 0; i < (int)s.size(); i++) {
    if (i > r)
      pal[i] = 1 - add;
    else
      pal[i] = min(r - i + 1, pal[l + r - i + add]);
    while (i - pal[i] - add >= 0 && i + pal[i] < (int)s.size() && s[i - pal[i] - add] == s[i + pal[i]])
      pal[i]++;
    pal[i]--;
    if (i + pal[i] > r) {
      r = i + pal[i] - add;
      l = i - pal[i];
    }
  }
  return pal;
}

int main() {
  in.tie(NULL);
  out.tie(NULL);
  ios_base::sync_with_stdio(false);
  string s; in >> s;
  auto pal = Manacher(s, 0), pal2 = Manacher(s, 1);
  ll ans = (ll)0;
  for (int x : pal)
    ans += 1 + x;
  for (int x : pal2)
    ans += x + 1;
  out << ans << "\n";
  return 0;
}