Cod sursa(job #894547)

Utilizator Al3ks1002Alex Cociorva Al3ks1002 Data 26 februarie 2013 21:57:12
Problema PScPld Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.68 kb
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int SMAX=1000005;
int n,i,j,P[SMAX*2],l,r,N,maxr,center;
long long sol;
char a[SMAX],S[SMAX*2];
int main()
{
    freopen("pscpld.in","r",stdin);
    freopen("pscpld.out","w",stdout);
    fgets(a+1,SMAX,stdin); n=strlen(a+1)-1;
    for(i=1;i<=n;i++) {S[++N]=a[i]; P[N++]=1;}
    for(i=1;i<=N-1;i++)
    {
        if(i<=maxr) P[i]=min(P[2*center-i],maxr-i+1);
        l=i; r=i; l--; r++; l-=P[i]; r+=P[i];
        while(l>=1 && r<=N && S[l]==S[r]) {P[i]+=2;l-=2;r+=2;}
        r-=2; if(r>maxr) {maxr=r; center=i;}
        sol+=(P[i]+1)/2;
    }
    printf("%lld\n",sol);
    return 0;
}