Cod sursa(job #2417340)

Utilizator stelian2000Stelian Chichirim stelian2000 Data 29 aprilie 2019 16:00:04
Problema PScPld Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <bits/stdc++.h>

using namespace std;

char s1[1000010],s[1000010];
int d[2000010];

int main()
{
    freopen("pscpld.in","r",stdin);
    freopen("pscpld.out","w",stdout);
    int n=0;
    scanf("%s",s1+1);
    int l=strlen(s1+1);
    for(int i=1;i<=l;i++)
    {
        s[++n]=s1[i];
        s[++n]='*';
    }
    n--;
    int lmax=0,poz=0;
    for(int i=1;i<=n;i++)
    {
        if(i+d[i]<poz+lmax) d[i]=min(poz+lmax-i,d[2*poz-i]);
        if(i+d[i]>=poz+lmax) while(i+d[i]<n && i-d[i]>1 && s[i+d[i]+1]==s[i-d[i]-1]) d[i]++;
        if(i+d[i]>poz+lmax) {poz=i;lmax=d[i];}
    }
    long long ans=0;
    for(int i=1;i<=n;i++)
        if(s[i]=='*') ans+=(d[i]+1)/2;
        else ans+=d[i]/2+1;
    printf("%lld",ans);
    return 0;

}