Cod sursa(job #3041378)

Utilizator IvanAndreiIvan Andrei IvanAndrei Data 31 martie 2023 13:31:36
Problema PScPld Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>
#include <string>

using namespace std;

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

const long long max_size = 2e6 + 1;

long long p[max_size];

signed main ()
{
    string s, a;
    in >> a;
    s += '$';
    for (long long i = 0; i < a.size(); i++)
    {
        s += a[i];
        s += '$';
    }
    long long n = s.size() - 1, poz = 0, dr = 0, ans = 0;
    for (long long i = 1; i <= n; i++)
    {
        long long x = 2 * poz - i;
        if (dr > i)
        {
            p[i] = min(dr - i, p[x]);
        }
        while (i >= p[i] + 1 && i + p[i] - 1 <= n && s[i - p[i] - 1] == s[i + p[i] + 1])
        {
            p[i]++;
        }
        if (i + p[i] > dr)
        {
            poz = i;
            dr = i + p[i];
        }
        if (i % 2 == 1)
        {
            ans += (p[i] + 1) / 2;
        }
        else
        {
            ans += p[i] / 2;
        }
    }
    out << ans;
    in.close();
    out.close();
    return 0;
}