Cod sursa(job #1961875)

Utilizator BugirosRobert Bugiros Data 11 aprilie 2017 13:56:16
Problema Reguli Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.43 kb
#include <cstdio>
using namespace std;

const int MAXN = 500005;

long long v[MAXN];
int n;

int prefix[MAXN];

void citire()
{
    freopen("reguli.in","r",stdin);
    freopen("reguli.out","w",stdout);
    scanf("%d",&n);
    for (int i = 1;i <= n;++i)
        scanf("%lld",&v[i]);
    for (int i = 1;i < n;++i)
        v[i] = v[i + 1] - v[i];
    --n;
}



void creare_prefix()
{
    int k = 0;
    prefix[1] = 0;
    for (int i = 2;i <= n;++i)
    {
        while(k > 0 && v[k + 1] != v[i])
            k = prefix[k];
        if (v[k + 1] == v[i])
            ++k;
        prefix[i] = k;
    }
}

int l;


//sufix[i] - adevarat daca sufixul de lungime i este si prefix
bool sufix[MAXN];

void calculare_sufix()
{
    for (int k = prefix[n];k != 0;k = prefix[k])
        sufix[k] = true;
}

bool ok()
{
    int c = n / l;
    int r = n % l;
    if (prefix[n - r] == 0)
        return false;
    if ((n - r) % (n - r - prefix[n - r]) != 0)
        return false;
    if ((n - r) / (n - r - prefix[n - r]) != c)
        return false;
    if (!sufix[r])
        return false;
    return true;
}

void prelucrare()
{
    l = 1;
    while(!ok())
        ++l;
}

void afisare()
{
    printf("%d\n", l);
    for (int i = 1;i <= l;++i)
        printf("%d\n",v[i]);
}

int main()
{
    citire();
    creare_prefix();
    calculare_sufix();
    prelucrare();
    afisare();
    return 0;
}