luogu P4238 【模板】多项式求逆

背景：

多项式全家桶$\text{eating...}$

题目传送门：

https://www.luogu.org/problemnew/show/P4238

题意：

给定一个多项式$F(x)$，求一个多项式$G(x)$使得$F(x)\ast G(x)\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)$。

思路：

假设我们已经搞定了$F\ast A\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}{x}^{\lceil \frac{n}{2}\rceil })$。
那么有：
$$F(G-A)\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}{x}^{\lceil \frac{n}{2}\rceil })$$

因为$F̸=0$，所以有：
$$G-A\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}{x}^{\lceil \frac{n}{2}\rceil })$$

两边同时平方，得：
$$G^2+A^2-2AG\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)$$

左右同时乘上$F$，得：
$$\begin{array}{rl}G+A^{2}F-2A& \equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)\\ G& \equiv 2A-A^{2}F(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)\end{array}$$

用$\text{NTT}$优化。
最后倍增求解，易证得时间复杂度为：$\Theta (n\log n)$。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
const LL mod=998244353,G=3,inv_G=332748118;
using namespace std;
	LL a[300010],b[300010],f[300010],g[300010];
	int limit,n,l,r[300010];
LL dg(LL x,LL k)
{
	if(!k) return 1;
	LL op=dg(x,k>>1);
	if(k&1) return op*op%mod*x%mod; else return op*op%mod;
}
LL inv(LL x)
{
	return dg(x,mod-2);
}
void NTT(LL *now,int limit,int op)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		LL wn=dg(op==1?G:inv_G,(mod-1)/(mid<<1));
		for(int j=0;j<limit;j+=(mid<<1))
		{
			LL w=1;
			for(int k=0;k<mid;k++,w=(w*wn)%mod)
			{
				LL x=now[j+k],y=w*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
		}
	}
}
void work(int n,LL *f,LL *g)
{
	if(n==1)
	{
		g[0]=inv(f[0]);
		return;
	}
	work((n+1)>>1,f,g);
	limit=1,l=0;
	while(limit<(n<<1))
		limit<<=1,l++;
	for(int i=1;i<limit;i++)
	r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	for(int i=0;i<n;i++)
		a[i]=f[i],b[i]=g[i];
	NTT(a,limit,1);
	NTT(b,limit,1);
	for(int i=0;i<limit;i++)
		b[i]=b[i]*((2ll-a[i]*b[i]%mod+mod)%mod)%mod;
	NTT(b,limit,-1);
	LL INV=inv(limit);
	for(int i=0;i<n;i++)
		g[i]=b[i]*INV%mod;
}
int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%lld",&f[i]);
	work(n,f,g);
	for(int i=0;i<n;i++)
		printf("%lld ",g[i]);
}