Cod sursa(job #2769319)

Utilizator Y.MalmsteenB.P.M. Y.Malmsteen Data 14 august 2021 17:42:56
Problema Pascal Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <fstream>

using namespace std;

ifstream f("pascal.in");
ofstream g("pascal.out");

int R, D,
    fp[3], ex[3], nf;

void descfp()
{
    if(D == 4)
    {
        nf = 1;
        fp[1] = 2;
        ex[1] = 2;
    }
    else //if(D == 6)
    {
        nf = 2;
        fp[1] = 2;
        fp[2] = 3;
        ex[1] = ex[2] = 1;
    }
}

int expp(int n, int p)
{
    int e = 0;
    while(n % p == 0)
    {
        e++;
        n /= p;
    }
    return e;
}

int divid(int k)
{
    int ok = 1;
    for(int i = 1; i <= nf; i++)
    {
        ex[i] -= expp(R - k + 1, fp[i]) - expp(k, fp[i]);
        if(ex[i] > 0) ok = 0;
    }
    return ok;
}

int cLucas(int n, int p)
{
    int rez = n + 1, prod = 1;
    while(n > 0)
    {
        prod *= (n % p + 1);
        n /= p;
    }
    rez -= prod;
    return rez;
}

int main()
{
    int nrd = 0;
    f >> R >> D;
    if(R >= 2)
    {
        if(D == 2 || D == 3 || D == 5)
            nrd = cLucas(R, D);
        else
        {
            descfp();
            int r2 = (R + 1) / 2;
            for(int k = 1; k < r2; k++)
                nrd += divid(k);
            nrd *= 2;
            if(R % 2 == 0)
                nrd += divid(r2);
        }
    }
    g << nrd;
    f.close();
    g.close();
    return 0;
}