Cod sursa(job #2203865)

Utilizator Luca19Hritcu Luca Luca19 Data 13 mai 2018 15:30:47
Problema Numarare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.6 kb
#include<fstream>

using namespace std;

ifstream f("numarare.in");
ofstream g("numarare.out");

int i,p,sol,n,a[100010],v[100010];

int main()
{
    f>>n>>v[1];

    for(i=2;i<=n;++i) {
        f>>v[i];
        v[i-1]-=v[i];
    }
    p=1;
    --n;
    for(i=1;i<=n;++i) {
        if(i<p+a[p])
        a[i]=min(a[2*p-i],p+a[p]-i);
        if(i+a[i]>=p+a[p])
        {
            p=i;
            while(i-a[i]-1>=1&&i+a[i]+1<=n&&v[i-a[i]-1]==v[i+a[i]+1])
            ++a[i];
        }
    }
    sol=n;
    for(i=1;i<=n;++i)
        sol+=a[i];
    g<<sol<<'\n';
    return 0;
}