Cod sursa(job #218569)

Utilizator mircea_infoSuciu Mircea-Gabriel mircea_info Data 2 noiembrie 2008 16:24:11
Problema Reguli Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.54 kb
#include <cstdio>
#define MAXIMUS 500001

using namespace std;

int n,poz[MAXIMUS],dif[MAXIMUS],ok,sir[MAXIMUS];

void pref()
{
    int k=0;
    for(int i=2;i<n;i++)
    {
        while (k && dif[k+1]!=dif[i])
            k=poz[k];
        if(dif[k+1]==dif[i])
            k++;
        poz[i]=k;
    }
}

int eok(int x)
{
    int aux=poz[n];
    while(poz[aux]>x)
    {
        aux=poz[aux];
    }
    if(poz[aux]==x)
            return 1;
    return 0;
}

void solve()
{
    n--;
    int r,c;
    for(int l=1;l<=n;l++)
    {
        r=n%l;
        c=n/l;
        if(poz[n-r]>0 && ((n-r)%(n-r-poz[n-r])==0) && ((n-r)/(n-r-poz[n-r])==c) && eok(r))
        {
            printf("%d\n",l);
            for(int i=1;i<=l;i++)
                printf("%d\n",dif[i]);
            ok=1;
            return;
        }
    }
}

int check()
{
    for(int i=1;i<=n;i++)
        if(poz[i])
            return 1;
    return 0;
}

int main()
{
    freopen("reguli.in","r",stdin);
    freopen("reguli.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&sir[i]);
        if(i>1)
            dif[i-1]=sir[i]-sir[i-1];
    }
    pref();
    if(check())
    {
        solve();
        if(ok==0)
        {
            printf("%d\n",n);
            for(int i=1;i<=n;i++)
                printf("%d\n",sir[i]);
        }
    }
    else
    {
        printf("%d\n",n-1);
        for(int i=1;i<n;i++)
            printf("%d\n",dif[i]);
    }
    fclose(stdout);
    return 0;
}