Cod sursa(job #2054847)

Utilizator eduardandrei20Nechifor Eduard Andrei eduardandrei20 Data 2 noiembrie 2017 16:45:14
Problema Kperm Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>
using namespace std;

const int MODULO = 666013;
const int MAXN = 5010;

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



int n,k,c,r;

long long fact[MAXN];

long long putere(int a , int b)
{
    long long p = 1 ;

    while (b){

        if(b & 1) {p*=a,--b; p = p % MODULO;}
        a*=a;
        a=a%MODULO;
        b>>=1;

    }
    return p%MODULO;


}

int main()
{


    in >> n >> k;
    c = n / k;
    r = n % k;
    if (k % 2 == 0)//imposibil de facut k-permutare
    {
        out << 0 << '\n';
        return 0;
    }

    //altfel raspunsul este r! * (k - r)! * [(c + 1)!] ^ r * (c!) ^ (k - r)

    //ne trebuie factorialele
    fact[0] = 1;
    for (int i = 1;i <= n;++i)
        fact[i] = (fact[i - 1] * i) % MODULO;

    long long rasp = fact[r];
    rasp = (rasp * fact[k - r]) % MODULO;
    rasp = (rasp * putere(fact[c + 1], r)) % MODULO;
    rasp = (rasp * putere(fact[c],k - r)) % MODULO;

    out << rasp << '\n';
    return 0;
}