NTT快速数论变换模板

最新推荐文章于 2022-10-18 19:08:03 发布

Azcii

最新推荐文章于 2022-10-18 19:08:03 发布

阅读量189

点赞数 1

分类专栏： # 数学-多项式文章标签： NTT 多项式模板

本文链接：https://blog.csdn.net/qq_39565901/article/details/115455939

版权

数学-多项式专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何使用快速傅里叶变换(FFT)算法进行大规模整数乘法。通过实例展示了FFT如何高效地计算两个长整数的乘积，详细解释了NTT（Numeral Transformations Theory）过程，并提供了ksm函数来计算幂。在代码实现中，还演示了如何预处理输入数组以及逆向位移，最终输出乘法结果。

摘要由CSDN通过智能技术生成

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define md 998244353
#define N 4000010
#define ll long long
ll a[N], b[N];
int ns, p, rev[N];
ll ksm(ll x, ll y) {
	if(!y) return 1;
	ll l = ksm(x, y / 2);
	if(y % 2) return l * l % md * x % md;
	return l * l % md;
}
void ntt(ll *a, int r) {
	for(int i = 0; i < ns; i++) if(i < rev[i]) swap(a[i], a[rev[i]]);
	for(int i = 2; i <= ns; i *= 2) {
		ll t = ksm(3, (md - 1) / i);
		if(r == -1) t = ksm(t, md - 2);
		for(int j = 0; j < ns; j += i) {
			ll s = 1;
			for(int k = 0; k < i / 2; k++) {
				ll A = a[j + k], B = a[j + i / 2 + k] * s % md;
				a[j + k] = (A + B) % md;
				a[j + i / 2 + k] = (A - B + md) % md;
				s = s * t % md;
			}
		}
	}
	if(r == -1) {
		ll t = ksm(ns, md - 2);
		for(int i = 0; i < ns; i++) a[i] = a[i] * t % md;
	}
}
ll read() {
	int s = 0;
	char x = getchar();
	while(x < '0' || x > '9') x = getchar();
	while(x >= '0' && x <= '9') s = s * 10 + x - 48, x = getchar();
	return s;
}
int main() {
	int n, m, i;
	scanf("%d%d", &n, &m);
	for(i = 0; i <= n; i++) a[i] = read();
	for(i = 0; i <= m; i++) b[i] = read();
	ns = 1, p = 0;
	while(ns <= n + m) ns = ns * 2, p++;
	rev[0] = 0;
	for(i = 0; i < ns; i++) {
		rev[i] = rev[i / 2] / 2 + (i % 2) * ns / 2;
	}
	ntt(a, 1);
	ntt(b, 1);
	for(i = 0; i < ns; i++) a[i] = a[i] * b[i] % md;
	ntt(a, -1);
	for(i = 0; i <= n + m; i++) printf("%lld ", a[i]);
	return 0;
}