PS

백준 11051 이항계수 2

jyheo98 2020. 8. 4. 06:43

자주 쓰게 될 코드같다. 

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
vector<llong> fact;
 
llong add(llong a, llong b) {
    a += b;
    if (a >= MOD) a -= MOD;
    return a;
}
llong sub(llong a, llong b) {
    a -= b;
    if (a < 0) a += MOD;
    return a;
}
llong mul(llong a, llong b) {
    return (a * b) % MOD;
}
llong bin_pow(llong a, llong p) {
    if (p == 0return 1;
    if (p & 1return mul(a, bin_pow(a, p - 1));
    return bin_pow(mul(a, a), p / 2);
}
llong rev(llong x) {
    return bin_pow(x, MOD - 2);
}
llong c(llong n, llong k) {
    return mul(fact[n], mul(rev(fact[k]), rev(fact[n - k])));
}
 
int main()
{
    IOS;
    
    llong n, k;
    cin >> n >> k;
    fact = vector<llong>(n+1);
    fact[0= 1;
    for (int i = 1; i <= n; i++) {
        fact[i] = mul(fact[i - 1], i);
    }
    cout << c(n, k);
 
    return 0;
}
cs