Cod sursa(job #3137764)

Utilizator LucaMuresanMuresan Luca Valentin LucaMuresan Data 14 iunie 2023 19:47:24
Problema Pascal Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.57 kb
#include <bits/stdc++.h>
#pragma GCC optimize ("fast-math", "unroll-loops")

using namespace std;

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

const int NMAX = 5e6;

int prec[3][NMAX + 5];
const int prime[3] = {2, 3, 5};

int main()
{
    int n, d;
    in >> n >> d;
    for (int i=1; i<=n; i++)
    {
        int x = i;
        for (int j=0; j<3; j++)
        {
            prec[j][i] = prec[j][i-1];
            while (x % prime[j] == 0)
                x /= prime[j], prec[j][i]++;
        }
    }

    int ans = 0;

    if (d == 2)
    {
        for (int i=1; i<=n; i++)
        {
            if (prec[0][n] - prec[0][i] - prec[0][n-i])
                ans++;
        }
        out << ans;
    }
    else if (d == 3)
    {
        for (int i=1; i<=n; i++)
        {
            if (prec[1][n] - prec[1][i] - prec[1][n-i])
                ans++;
        }
        out << ans;
    }
    else if (d == 5)
    {
        for (int i=1; i<=n; i++)
        {
            if (prec[2][n] - prec[2][i] - prec[2][n-i])
                ans++;
        }
        out << ans;
    }
    else if (d == 4)
    {
        assert(0 == 1);
        for (int i=1; i<=n; i++)
        {
            if (prec[0][n] - prec[0][i] - prec[0][n-i] > 1)
                ans++;
        }
        out << ans;
    }
    else
    {
        for (int i=1; i<=n; i++)
        {
            if (prec[0][n] - prec[0][i] - prec[0][n-i] && prec[1][n] - prec[1][i] - prec[1][n-i])
                ans++;
        }
        out << ans;
    }

    return 0;
}