Cod sursa(job #2528891)

Utilizator NoemikulcsarKulcsar Noemi Noemikulcsar Data 22 ianuarie 2020 18:58:23
Problema PScPld Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <fstream>
#include <cstring>

using namespace std;

const int maxi=2000005;

ifstream cin("pscpld.in");
ofstream cout("pscpld.out");

long long mij,sum,v[2*maxi],n,st,jdr,jst,dr,k=-1;
char c[2*maxi];
string s;

int main()
{
    cin>>s;
    n=s.size();
    n*=2;
    for(int i=0;i<=n;i+=2){
        c[i]='*';
        c[i+1]=s[++k];
    }
    for(int i=1;i<=2*n;i++){
        if(i<=dr){
            mij=st+dr-i;
            v[i]=min(v[mij],mij-st+1);
            int jdr=i+v[i],jst=i-v[i];
            while(jst>=0 && jdr<=n && c[jdr]==c[jst]){
                jdr++;
                jst--;
                v[i]++;
            }
            if(dr<i+v[i]-1){
                dr=i+v[i]-1;
                st=i-v[i]+1;
            }
        }
        else{
            int jdr=i,jst=i;
            while(jst>=0 && jdr<=n && c[jdr]==c[jst]){
                jdr++;
                jst--;
                v[i]++;
            }
            dr=i+v[i]-1;
            st=i-v[i]+1;
        }
    }
    for(int i=1;i<=n;i++)
        sum+=v[i]/2;
    cout<<sum;
    return 0;

}