summaryrefslogtreecommitdiff
path: root/math/binomial0.cpp
diff options
context:
space:
mode:
authormzuenni <michi.zuendorf@gmail.com>2023-05-08 11:25:47 +0200
committermzuenni <michi.zuendorf@gmail.com>2023-05-08 11:25:47 +0200
commit558aeeed893c98bf1851856c4effd1ac4c3722e7 (patch)
treec7adc9830e0313bc3dccf3caf4764bbe3a6ca13f /math/binomial0.cpp
parent799d0212b6cd222ab78ba3a4a63f867f1bb9c143 (diff)
improved binomial
Diffstat (limited to 'math/binomial0.cpp')
-rw-r--r--math/binomial0.cpp14
1 files changed, 14 insertions, 0 deletions
diff --git a/math/binomial0.cpp b/math/binomial0.cpp
new file mode 100644
index 0000000..d9af917
--- /dev/null
+++ b/math/binomial0.cpp
@@ -0,0 +1,14 @@
+constexpr ll lim = 10'000'000;
+ll fac[lim], inv[lim];
+
+void precalc() {
+ fac[0] = inv[0] = 1;
+ for (int i = 1; i < lim; i++) fac[i] = fac[i-1] * i % mod;
+ inv[lim - 1] = multInv(fac[lim - 1], mod);
+ for (int i = lim - 1; i > 0; i--) inv[i-1] = inv[i] * i % mod;
+}
+
+ll calc_binom(ll n, ll k) {
+ if (n < 0 || n < k || k < 0) return 0;
+ return (fac[n] * fac[n-k] % mod) * fac[k] % mod;
+}