多项式求逆

问题：
求多项式(g(x))，满足(f(x)*g(x)equiv 1pmod{x^n})。

Q1:这是什么意思？

A1：即(f(x)*g(x))最后只有常数项系数为(1)，其余系数都为(0)。

Q2：为什么要求模(x^n)？

A2:这是为了将(n)次方以上的项全都除去，否则(g(x))会有无穷多项。

用倍增的方法可以(O(nlog n))求出。

假设我们求出了(h(x))，满足(f(x)*h(x)equiv 1pmod{x^{lceilfrac{n}{2} ceil}})。

我们要求(g(x))，满足(f(x)*g(x)equiv 1pmod{x^n})。

进行推导：
(f(x)*h(x)equiv 1pmod{x^{lceilfrac{n}{2} ceil}})
(f(x)*g(x)equiv 1pmod{x^n})
(f(x)*g(x)equiv f(x)*h(x)pmod{x^{lceilfrac{n}{2} ceil}})（为什么正确需要考虑模的意义：忽略更高次项）
(f(x)*(g(x)-h(x))equiv 0pmod{x^{lceilfrac{n}{2} ceil}})
(g(x)-h(x)equiv 0pmod{x^{lceilfrac{n}{2} ceil}})
((g(x)-h(x))^2equiv 0pmod{x^{lceilfrac{n}{2} ceil}})
(g(x)^2-2*g(x)*h(x)+h(x)^2equiv 0pmod{x^{lceilfrac{n}{2} ceil}})
此时有个结论：
(g(x)^2-2*g(x)*h(x)+h(x)^2equiv 0pmod{x^n})

因为平方前((g(x)-h(x)))中([0,lceilfrac{n}{2} ceil-1])的系数都为(0)，考虑(iin[lceilfrac{n}{2} ceil,n-1])，(C_i=sumlimits_{j<=i}A_j*B_{i-j})，其中(j,i-j)中必定有一个属于([0,lceilfrac{n}{2} ceil-1])，因此(C_i=0)。

两边同乘(f(x))：
(g(x)-2*h(x)+f(x)h(x)^2equiv 0pmod{x^n})
(g(x)equiv2*h(x)-f(x)*h(x)^2pmod{x^n})
(g(x)equiv h(x)*(2-f(x)*h(x))pmod{x^n})

于是可以递归分治求了。

模板题

code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=4*1e5+10;
const ll mod=998244353;
const ll g=3;
const ll invg=332748118;
int n,lim,len;
int pos[maxn];
ll a[maxn],b[maxn],c[maxn];
inline ll read()
{
	char c=getchar();ll res=0,f=1;
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')res=res*10+c-'0',c=getchar();
	return res*f; 
}
inline ll power(ll x,ll k)
{
	ll res=1;
	while(k)
	{
		if(k&1)res=res*x%mod;
		x=x*x%mod;k>>=1;
	}
	return res;
}
void NTT(ll* a,int op)
{
	for(int i=0;i<lim;i++)if(i<pos[i])swap(a[i],a[pos[i]]);
	for(int l=1;l<lim;l<<=1)
	{
		ll wn=power(op==1?g:invg,(mod-1)/(l<<1));
		for(int i=0;i<lim;i+=l<<1)
		{
			ll w=1;
			for(int j=0;j<l;j++,w=w*wn%mod)
			{
				ll x=a[i+j],y=w*a[i+l+j];
				a[i+j]=(x+y)%mod;a[i+l+j]=((x-y)%mod+mod)%mod;
			}
		}
	}
	if(op==1)return;
	ll inv=power(lim,mod-2);
	for(int i=0;i<lim;i++)a[i]=a[i]*inv%mod;
}
void solve(ll* a,ll* b,int n)
{
	if(n==1){b[0]=power(a[0],mod-2);return;}
	solve(a,b,(n+1)>>1);
	lim=1,len=0;
	while(lim<(n<<1))lim<<=1,len++;
	for(int i=0;i<lim;i++)pos[i]=(pos[i>>1]>>1)|((i&1)<<(len-1));
	for(int i=0;i<n;i++)c[i]=a[i];
	for(int i=n;i<lim;i++)c[i]=0;
	NTT(c,1);NTT(b,1);
	for(int i=0;i<lim;i++)b[i]=((2ll-b[i]*c[i]%mod)%mod+mod)%mod*b[i]%mod;
	NTT(b,-1);
	for(int i=n;i<lim;i++)b[i]=0;
}
int main()
{
	n=read();
	for(int i=0;i<n;i++)a[i]=(read()%mod+mod)%mod;
	solve(a,b,n);
	for(int i=0;i<n;i++)printf("%lld ",(b[i]%mod+mod)%mod);
	return 0;
}