summaryrefslogtreecommitdiff
path: root/math/transforms/multiplyNTT.cpp
diff options
context:
space:
mode:
authorNoobie99 <noob999noob999@gmail.com>2024-02-05 22:23:53 +0100
committerNoobie99 <noob999noob999@gmail.com>2024-02-05 22:23:53 +0100
commitf4dafa64e27e63d8cde5adcfbfa98e2a30fb7504 (patch)
tree936fc0e98ac7e1ff6222b7657fb4a6b90d798080 /math/transforms/multiplyNTT.cpp
parent631335a9c1dd0fba0e17977253d0a4b033aee59a (diff)
renamed ntt and bitwiseConv function + more multiply functions
Diffstat (limited to 'math/transforms/multiplyNTT.cpp')
-rw-r--r--math/transforms/multiplyNTT.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/math/transforms/multiplyNTT.cpp b/math/transforms/multiplyNTT.cpp
new file mode 100644
index 0000000..806d124
--- /dev/null
+++ b/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;
+}