Cod sursa(job #2603878)

Utilizator robert.barbu27robert barbu robert.barbu27 Data 21 aprilie 2020 08:56:37
Problema PScPld Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.89 kb
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 1000000

using namespace std;

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

long long v[2*NMAX+10], sol;
string s;
vector <char> t;

int main()
{
    f >> s;
    t.push_back('&');
    int m = s.size();
    for(long long i=0; i<m; i++)
        {   t.push_back('#');
            t.push_back(s[i]);
        }
    t.push_back('#');
    t.push_back('%');
    long long C = 0, R = 0, n = t.size();
    for(int i=1; i<n-1; i++)
        {   long long ogl = 2 * C - i;
            if(i <= R) v[i] = min(R-i, v[ogl]);
            while(t[i+v[i]+1] == t[i-v[i]-1]) v[i]++;
            if(i + v[i] > R)
                {   C = i;
                    R = v[i] + i;
                }
        }
    for(int i=0; i<n; i++)
        if(v[i] > 0) sol = sol + (v[i] - 1) / 2 + 1;
    g << sol << '\n';
    return 0;
}