Cod sursa(job #3041047)

Utilizator IvanAndreiIvan Andrei IvanAndrei Data 30 martie 2023 21:11:35
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 v, s;
    in >> v;
    s += '$';
    for (long long i = 0; i < v.size(); i++)
    {
        s += v[i];
        s += '$';
    }
    long long n = v.size() * 2, 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;
}