Cod sursa(job #2773053)

Utilizator mjmilan11Mujdar Milan mjmilan11 Data 4 septembrie 2021 13:04:12
Problema PScPld Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("pscpld.in");
ofstream fout("pscpld.out");
const int NMAX = 1000005;
char c[NMAX],a[2*NMAX];
int n,P[NMAX],rasp=0;
void citire(){
    fin >> (c+1);
    n=strlen(c+1);
    for(int i=1;i<=n;i++){
        a[i*2-1]='#';
        a[i*2]=c[i];
    }
    a[n*2+1]='#';
    n=2*n+1;
}
void pscpld(){
    int c=0;
    for(int i=1;i<=n;i++){
        if(i<=c+P[c]) P[i]=min(P[2*c-i],c+P[c]-i);
        while(a[i+P[i]+1]==a[i-P[i]-1]) P[i]++;
        if(i+P[i]>c+P[c]){
            c=i;
        }
    }
}
void solve(){
    for(int i=1;i<=n;i++){
        if(i%2==0) rasp+=(P[i]+1)/2;
        else       rasp+=P[i]/2;
    }
}
void afis(){
    fout << rasp;
}
int main()
{
    citire();
    pscpld();
    solve();
    afis();
    return 0;
}