Cod sursa(job #1976338)

Utilizator georgerapeanuRapeanu George georgerapeanu Data 3 mai 2017 09:37:15
Problema PScPld Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
FILE *f=fopen("pscpld.in","r");
FILE *g=fopen("pscpld.out","w");
char A[1000005];
char B[2000005];
int L[2000005];
int N;
int id,i;
long long rez;
int main()
{
    fgets(A+1,1000005,f);
    N=strlen(A+1);N=N-(A[N]=='\n');
    for(int i=1;i<=N;i++)
    {
        B[2*i-1]='#';
        B[2*i]=A[i];
    }
    B[N=2*N+1]='#';
    id=0;
    for(int i=1;i<=N;i++)
    {
        if(id+L[id]>=i)L[i]=min(L[2*id-i],id+L[id]-i);
        else L[i]=0;
        while(i-L[i]>1&&i+L[i]<N&&B[i-L[i]-1]==B[i+L[i]+1]) L[i]++;
        if(i+L[i]>id+L[id])id=i;
        rez+=(1+L[i])/2;
    }
    fprintf(g,"%lld",rez);
    fclose(f);
    fclose(g);
    return 0;
}