Cod sursa(job #2841456)

Utilizator betybety bety bety Data 29 ianuarie 2022 19:14:07
Problema PScPld Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("pscpld.in");
ofstream out("pscpld.out");
vector<int> manacher(string s)
{
    string a="@#";
    for(char c:s)
        a+=c+string("#");
    a+='$';
    int l=0,r=-1;
    int n=a.size()-2;
    vector<int> p(n+2);
    for(int i=1;i<=n;++i)
    {
        p[i]=max(0,min(r-i,p[l+r-i]));
        while(a[i-p[i]]==a[i+p[i]])
            ++p[i];
        if(i+p[i]>r)
            l=i-p[i],r=i+p[i];
    }
    return vector<int>(begin(p)+2,end(p)-2);
}
string s;
int main()
{
    in>>s;
    long long ans=0;
    vector<int> v=manacher(s);
    for(int x:v) ans+=x/2; out<<ans<<'\n';
    return 0;
}