From 915bbaea690d3d352b06d5a2660166de7fd9cd21 Mon Sep 17 00:00:00 2001
From: mzuenni <michi.zuendorf@gmail.com>
Date: Sat, 17 Aug 2024 20:07:14 +0200
Subject: improve series operations

---
 content/math/transforms/seriesOperations.cpp | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

(limited to 'content/math')

diff --git a/content/math/transforms/seriesOperations.cpp b/content/math/transforms/seriesOperations.cpp
index fc36f1e..b405698 100644
--- a/content/math/transforms/seriesOperations.cpp
+++ b/content/math/transforms/seriesOperations.cpp
@@ -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) {
-- 
cgit v1.2.3