【模板】多項式乘法逆

題目連結：luogu P4238

題目大意

給你一個多項式 F(x)，要你求一個多項式 G(x)，使得 F(x)*G(x)≡1(mod x^n)，係數對 998244353 取模。

思路

考慮用倍增的方法。
首先只有 \(1\) 項的時候，就直接是逆元。

然後我們考慮已知 \(0\sim \frac{n}{2}\) 項的答案 \(G'(x)\)，然後求 \(0\sim n\) 的答案 \(G(x)\)。（不難看出後面增加不會影響前面的）

那就有 \(G'(x)=G(x)\pmod{x^{\frac{n}{2}}}\)
即 \(G'(x)-G(x)=0\pmod{x^{\frac{n}{2}}}\)

我們考慮把 \(\pmod{x^{\frac{n}{2}}}\) 變成 \(\pmod{x^{n}}\)，用平方。
\((G'(x)-G(x))^2=0\pmod{x^{n}}\)
\(G'(x)^2-2G(x)G'(x)+G(x)^2=0\pmod{x^{n}}\)

然後每一項和右邊的 \(0\) 都乘上 \(F(x)\)：
\(F(x)G'(x)^2-2G'(x)+G(x)=0\pmod{x^{n}}\)
\(G(x)=2G'(x)-F(x)G'(x)^2\pmod{x^{n}}\)

然後就可以用多項式 NTT 來算右邊，然後直接 \(O(n)\) 加就好了。

然後好像有一個小小可以優化的方法：
你先算 \(F(x)G'(x)\)

然後由於我們只要 \(\frac{n}{2}\sim n-1\) 項的。
你第一次乘的時候前面 \(0\sim \frac{n}{2}-1\) 項都是 \(0\)（因為它們在 \(\bmod{x^{\frac{n}{2}}}\) 的時候是 \(0\)）
所以前面可以直接省去，然後第二次也可以直接省去前面的。

所以就不用每次都擴大多項式的範圍，就可以節省時間。

程式碼

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define mo 998244353

#define clear(f, n) memset(f, 0, n * sizeof(ll))
#define cpy(f, g, n) memcpy(f, g, n * sizeof(ll))

using namespace std;

int n, an[300001], limit, l_size;
ll f[300001], G, Gv;
ll w[300001], r[300001], tmp[300001];

ll ksm(ll x, ll y) {
	ll re = 1;
	while (y) {
		if (y & 1) re = re * x % mo;
		x = x * x % mo;
		y >>= 1;
	}
	return re;
}

void get_an() {
	for (int i = 0; i < limit; i++)
		an[i] = (an[i >> 1] >> 1) | ((i & 1) << (l_size - 1));
}

void NTT(ll *f, ll op) {//NTT
	get_an();
	for (int i = 0; i < limit; i++)
		if (i < an[i]) swap(f[i], f[an[i]]);
	
	for (int mid = 1; mid < limit; mid <<= 1) {
		ll Wn = ksm(op == 1 ? G : Gv, (mo - 1) / (mid << 1));
		for (int R = (mid << 1), j = 0; j < limit; j += R) {
			ll w = 1;
			for (int k = 0; k < mid; k++, w = w * Wn % mo) {
				ll x = f[j + k], y = w * f[j + mid + k] % mo;
				f[j + k] = (x + y) % mo;
				f[j + mid + k] = (x - y + mo) % mo;
			}
		}
	}
	
	if (op == -1) {//在這裡直接乘了
		ll liv = ksm(limit, mo - 2);
		for (int i = 0; i < limit; i++)
			f[i] = f[i] * liv % mo;
	}
}

void px(ll *x, ll *y) {
	for (int i = 0; i < limit; i++)
		x[i] = x[i] * y[i] % mo;
}

void cheng(ll *x, int n, ll *y, int m) {//只是寫著，並沒有用
	limit = 1; l_size = 0;
	while (limit < n + m + 1) {
		limit <<= 1; l_size++;
	}
	NTT(x, 1); NTT(y, 1);
	px(x, y); NTT(x, -1);
}

void invp(ll *F, int n) {
	w[0] = ksm(F[0], mo - 2);
	l_size = 0;
	for (int len = 2; (len >> 1) <= n; len <<= 1) {//倍增
		limit = len; l_size++;
		for (int i = 0; i < (len >> 1); i++) r[i] = w[i];
		cpy(tmp, F, len);//按著操作來把三個乘上
		NTT(tmp, 1); NTT(r, 1);
		px(r, tmp); NTT(r, -1);
		clear(r, (len >> 1));
		cpy(tmp, w, len);
		NTT(tmp, 1); NTT(r, 1);
		px(r, tmp); NTT(r, -1);
		
		for (int i = len >> 1; i < len; i++)//按著公式弄
			w[i] = (w[i] * 2 - r[i] + mo) % mo;
	}
	cpy(F, w, n);
	clear(tmp, n); clear(w, n); clear(r, n);//清空為好
}

int main() {
	G = 3; Gv = ksm(G, mo - 2);
	
	scanf("%d", &n);
	for (int i = 0; i < n; i++) scanf("%lld", &f[i]);
	
	invp(f, n);
	
	for (int i = 0; i < n; i++) printf("%lld ", f[i]);
	
	return 0;
}