summaryrefslogtreecommitdiff
path: root/content/math/transforms/seriesOperations.cpp
diff options
context:
space:
mode:
authorGloria Mundi <gloria@gloria-mundi.eu>2024-11-16 15:39:23 +0100
committerGloria Mundi <gloria@gloria-mundi.eu>2024-11-16 15:39:23 +0100
commit72bd993483453ed8ebc462f1a33385cd355d486f (patch)
treec5592ba1ed2fed79e26ba6158d097c9ceb43f061 /content/math/transforms/seriesOperations.cpp
parent98567ec798aa8ca2cfbcb85c774dd470f30e30d4 (diff)
parent35d485bcf6a9ed0a9542628ce4aa94a3326d0884 (diff)
merge mzuenni changes
Diffstat (limited to 'content/math/transforms/seriesOperations.cpp')
-rw-r--r--content/math/transforms/seriesOperations.cpp16
1 files changed, 9 insertions, 7 deletions
diff --git a/content/math/transforms/seriesOperations.cpp b/content/math/transforms/seriesOperations.cpp
index 4743674..b405698 100644
--- a/content/math/transforms/seriesOperations.cpp
+++ b/content/math/transforms/seriesOperations.cpp
@@ -4,7 +4,7 @@ vector<ll> poly_inv(const vector<ll>& a, int n) {
vector<ll> a2 = a, q2 = q;
a2.resize(2*len), q2.resize(2*len);
ntt(q2);
- for (int j : {0, 1}) {
+ for (int _ : {0, 1}) {
ntt(a2);
for (int i = 0; i < 2*len; i++) a2[i] = a2[i]*q2[i] % mod;
ntt(a2, true);
@@ -24,10 +24,13 @@ vector<ll> poly_deriv(vector<ll> a) {
}
vector<ll> poly_integr(vector<ll> a) {
- if (a.empty()) return {0};
- a.push_back(a.back() * powMod(sz(a), mod-2, mod) % mod);
- for (int i = sz(a)-2; i > 0; i--)
- a[i] = a[i-1] * powMod(i, mod-2, mod) % mod;
+ static vector<ll> inv = {0, 1};
+ for (static int i = 2; i <= sz(a); i++)
+ inv.push_back(mod - mod / i * inv[mod % i] % mod);
+
+ a.push_back(0);
+ for (int i = sz(a) - 1; i > 0; i--)
+ a[i] = a[i-1] * inv[i] % mod;
a[0] = 0;
return a;
}
@@ -35,8 +38,7 @@ vector<ll> poly_integr(vector<ll> a) {
vector<ll> poly_log(vector<ll> a, int n) {
a = mul(poly_deriv(a), poly_inv(a, n));
a.resize(n-1);
- a = poly_integr(a);
- return a;
+ return poly_integr(a);
}
vector<ll> poly_exp(vector<ll> a, int n) {