Cod sursa(job #1164177)

Utilizator MacWonkMihai Alexandru Cosmin MacWonk Data 1 aprilie 2014 21:51:05
Problema Numarare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.59 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;
}