Cod sursa(job #1164808)

Utilizator gapdanPopescu George gapdan Data 2 aprilie 2014 12:21:43
Problema Numarare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.68 kb
#include<fstream>
using namespace std;
int n,i,v[100005],poz[100005],p,sol;
inline int min(int a,int b)
{
    if (a<b) return a;
    else return b;
}
int main()
{
    ifstream f("numarare.in");
    ofstream g("numarare.out");
   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+poz[p]) poz[i]=min(poz[2*p-i],p+poz[p]-i);
        if(i+poz[i]>=p+poz[p])
        {
            p=i;
            while(i-poz[i]-1>=1 && i+poz[i]+1<=n && v[i-poz[i]-1]==v[i+poz[i]+1])
            ++poz[i];
        }
    }
    sol=n;
    for(i=1;i<=n;++i) sol+=poz[i];
    g<<sol<<'\n';
    return 0;
}