Cod sursa(job #1985252)

Utilizator Bodo171Bogdan Pop Bodo171 Data 27 mai 2017 12:17:32
Problema Numarare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <iostream>
#include <fstream>
using namespace std;
const int nmax=200005;
int s[nmax],sum[nmax],p[nmax];
long long tot;
int n,i,j,x,prc,spec,c,r;
int main()
{
    ifstream f("numarare.in");
    ofstream g("numarare.out");
    f>>n;
    f>>prc;spec=200001;
    for(i=2;i<=n;i++)
    {
        f>>x;
        s[2*(i-1)-1]=x-prc;
        s[2*(i-1)]=spec;
        prc=x;
    }
    s[0]=spec+1;
    for(i=1;i<=2*(n-1);i++)
    {
        sum[i]=sum[i-1];
        if(s[i]!=spec)
            sum[i]++;
    }
    for(i=1;i<=2*(n-1);i++)
        cout<<s[i]<<' ';
    cout<<'\n';
    for(i=1;i<=2*(n-1);i++)
    {
        if(i>r) j=0;
        else j=min(p[c*2-i],r-i);
        while(s[i-j]==s[i+j])
            j++;
        p[i]=j-1;
        if(i+p[i]>r)
            r=i+p[i],c=i;
        if(i%2)tot+=1LL*(sum[i]-sum[i-p[i]-1]);
    }
    g<<tot;
    return 0;
}