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 == 0) return 1;
if (p & 1) return 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 |