Cod sursa(job #2850386)

Utilizator puica2018Puica Andrei puica2018 Data 16 februarie 2022 18:25:25
Problema Numarare Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("numarare.in");
ofstream fout("numarare.out");

int n;
int a[100005],v[100005],l[100005];
long long ans=1;

int main()
{
    fin>>n;
    for(int i=1; i<=n; i++) fin>>a[i];
    if(n==1)
    {
        fout<<"0\n";
        return 0;
    }
    for(int i=1; i<n; i++)
        v[i]=a[i+1]-a[i];
    /// cate secvente palindromice de lungime impara are sirul v
    int c=1,r=1;
    for(int i=2; i<n; i++)
    {
        if(r>=i) l[i]=min(l[2*c-i],r-i);
        while(i+l[i]+1<=n && i-l[i]-1>=1 && v[i+l[i]+1]==v[i-l[i]-1]) l[i]++;
        if(i+l[i]>r)
        {
            r=i+l[i];
            c=i;
        }
        ans+=l[i]+1;
    }
    fout<<ans<<"\n";
    return 0;
}