-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathcombination_mod.cpp
77 lines (69 loc) · 1.75 KB
/
combination_mod.cpp
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
// iran
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
using vvi = vector<vi>;
constexpr i64 MOD = 1e9 + 7;
i64 modpow(i64 a, i64 n, i64 mod) {
if (n == 0) return 1;
if (n % 2 == 0) {
i64 t = modpow(a, n / 2, mod);
return t * t % mod;
}
return a * modpow(a, n - 1, mod) % mod;
}
i64 modinv(i64 a, i64 mod) {
// mod is prime
return modpow(a, mod - 2, mod);
}
struct Combination {
const int MAX_N = 1 << 21;
vi fact = vi(MAX_N);
vi factinv = vi(MAX_N);
Combination() {
fact[0] = 1;
for (int i = 1; i < MAX_N; i++) {
fact[i] = fact[i - 1] * i % MOD;
}
factinv[MAX_N - 1] = modinv(fact[MAX_N - 1], MOD);
for (int i = MAX_N - 2; i >= 0; i--) {
factinv[i] = factinv[i + 1] * (i + 1) % MOD;
}
}
i64 ncr(i64 n, i64 r) {
if (r < 0 || r > n) return 0;
return fact[n] * factinv[r] % MOD * factinv[n - r] % MOD;
}
i64 npr(i64 n, i64 r) {
if (r < 0 || r > n) return 0;
return fact[n] * factinv[n - r] % MOD;
}
i64 nhr(i64 n, i64 r) {
if (n == 0 && r == 0) return 1;
return ncr(n + r - 1, r);
}
};
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int t;
cin >> t;
Combination comb;
while (t--) {
char c;
i64 n, k;
cin >> c;
if (c == 'C') {
cin >> c >> n >> c >> k >> c;
cout << comb.ncr(n, k) << '\n';
} else if (c == 'P') {
cin >> c >> n >> c >> k >> c;
cout << comb.npr(n, k) << '\n';
} else {
cin >> c >> n >> c >> k >> c;
cout << comb.nhr(n, k) << '\n';
}
}
}