summaryrefslogtreecommitdiff
path: root/content/math/transforms/multiplyNTT.cpp
diff options
context:
space:
mode:
authorGloria Mundi <gloria@gloria-mundi.eu>2024-11-16 01:24:14 +0100
committerGloria Mundi <gloria@gloria-mundi.eu>2024-11-16 01:24:14 +0100
commit98567ec798aa8ca2cfbcb85c774dd470f30e30d4 (patch)
tree5113d5cc24d1ad5f93810b6442ce584a36950dc8 /content/math/transforms/multiplyNTT.cpp
parentad3856a6b766087df0036de0b556f4700a6498c9 (diff)
parent8d11c6c8213f46f0fa19826917c255edd5d43cb1 (diff)
mzuenni tests
Diffstat (limited to 'content/math/transforms/multiplyNTT.cpp')
-rw-r--r--content/math/transforms/multiplyNTT.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/content/math/transforms/multiplyNTT.cpp b/content/math/transforms/multiplyNTT.cpp
new file mode 100644
index 0000000..806d124
--- /dev/null
+++ b/content/math/transforms/multiplyNTT.cpp
@@ -0,0 +1,8 @@
+vector<ll> mul(vector<ll> a, vector<ll> b) {
+ int n = 1 << (__lg(sz(a) + sz(b) - 1) + 1);
+ a.resize(n), b.resize(n);
+ ntt(a), ntt(b);
+ for (int i=0; i<n; i++) a[i] = a[i] * b[i] % mod;
+ ntt(a, true);
+ return a;
+}