Cod sursa(job #1306588)

Utilizator gbibBacotiu Gabi gbib Data 31 decembrie 2014 11:19:26
Problema PScPld Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <fstream>
#include <cstring>
using namespace std;
ifstream in("pscpld.in");
ofstream out("pscpld.out");
char s[1000005],t[2000005];
int p[2000005];
int main()
{int r,c,i,j,lg,n;
long long sol=0;
in>>s;
lg=strlen(s);
t[0]='^';
for(i=0,j=1;i<=lg;i++)
{
    t[j++]='#';
    t[j++]=s[i];
}

t[j-1]='$';

n=strlen(t);
r=c=0;
for(i=1;i<n;i++)
{
    j=c-(i-c);
    if(r>i)
    {
        if(p[j]<r-i)
        {
            p[i]=p[j];
        }
        else
        {
            p[i]=r-i;
        }
    }
    while(t[i+1+p[i]]==t[i-1-p[i]])
    {
        p[i]++;
    }
    if(i+p[i]>r)
    {
        c=i;
        r=i+p[i];
    }
}

for(i=1;i<n;i++)
{
    if(i%2==1)
    sol+=p[i]/2;
    else
    sol+=(p[i]+1)/2;
}
out<<sol<<'\n';
    return 0;
}