多项式乘法逆

最新推荐文章于 2024-09-25 14:20:52 发布

原创最新推荐文章于 2024-09-25 14:20:52 发布 · 551 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法

洛谷 P4238 【模板】多项式乘法逆

给出多项式 $f (x)$ ，要你求多项式 $g (x)$ ，满足 $f(x)\cdot g(x)\equiv1\pmod{x^n}$ 。系数对 $998244353$ 取模。

分析

首先要明确，多项式对 $x^n$ 取模代表次数大于等于 $n$ 的那些项就不管了。

我们可以采用分治的思想来求多项式乘法逆。

如果 $f (x)$ 只有常数项，那么 $g (x)$ 就是 $f (x)$ 的逆元。

假设我们已经求出了 $f(x)\cdot G(x)\equiv1\pmod{x^{\lceil\frac n2\rceil}}$ ，要求函数 $g (x)$ 满足 $f(x)\cdot g(x)\equiv1\pmod{x^n}$ 。

现在的目标是想办法用 $G (x), f (x)$ 表示出 $g (x)$ 。

因为 $f(x)\cdot g(x)\equiv1\pmod{x^n}$ ，所以 $f(x)\cdot g(x)$ 中次数小于 $n$ 的项的系数除了常数项都为 $0$ ，那么对于次数小于 $\left\lceil\frac{n}{2}\right\rceil$ 的系数也一样，即
$f(x)\cdot g(x)\equiv1\pmod{x^{\lceil\frac n2\rceil}}$

与前面一个式子 $f(x)\cdot G(x)\equiv1\pmod{x^{\lceil\frac n2\rceil}}$ 相减得到
$f(x)\cdot\left(g(x)-G(x)\right)\equiv0\pmod{x^{\lceil\frac n2\rceil}}$

明显 $f(x)\not\equiv0\pmod{x^{\lceil\frac n2\rceil}}$ ，可得
$g(x)-G(x)\equiv0\pmod{x^{\lceil\frac n2\rceil}}$

这时， $g (x) - G (x)$ 最高系数不为零的项的次数最少都是 $x^{\lceil\frac n2\rceil}$ 了，所以对于 $g(x)-G(x))^2$ 次数小于 $n$ 的项的系数肯定都是零了。

所以
$g(x)^2+G(x)^2-2g(x)G(x)\equiv0\pmod{x^{n}}$

又因为 $f(x)\cdot g(x)\equiv1\pmod{x^n}$

两边同乘 $f (x)$ 得
$g(x)\equiv 2G(x)-f(x)G(x)^2\pmod{x^n}$

这样， $g (x)$ 就求出来了。

用 $NTT$ 求 $g (x)$ 时可以把 $G (x), f (x)$ 转换为点值表示法，算出 $g (x)$ 后再转回去。

时间复杂度的递推式 $T(n)=T(\dfrac n2)+O(n\log n)$ ，总的时间复杂度为 $n\log n)$ 。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=(1<<18)+1;
const ll mod=998244353,g=3;
int len=1,n;
ll a1[N],w,wn,a[N],ans[N];
ll ksm(ll a,ll b)
{
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void change(ll num[])
{
    for(int i=1,j=len/2;i<len-1;i++){
        if(i<j) swap(num[i],num[j]);
        int k=len/2;
        while(j>=k) j-=k,k>>=1;
        if(j<k) j+=k;
    }
}
void ntt(ll num[],int fl)
{
    for(int i=2;i<=len;i<<=1){
        if(fl==1) wn=ksm(g,(mod-1)/i);
        else wn=ksm(g,mod-1-(mod-1)/i);
        for(int j=0;j<len;j+=i){
            w=1;
            for(int k=j;k<j+i/2;k++){
                ll u=w*num[k+i/2]%mod,t=num[k];
                num[k]=(t+u)%mod;
                num[k+i/2]=(t-u+mod)%mod;
                w=w*wn%mod;
            }
        }
    }
    if(fl==-1){
        ll inv=ksm(len,mod-2);
        for(int i=0;i<len;i++) num[i]=num[i]*inv%mod;
    }
}
int read()
{
    int sum=0,c=getchar();
    while(c<48||c>57) c=getchar();
    while(c>=48&&c<=57) sum=sum*10+c-48,c=getchar();
    return sum;
}
void solve(int n,ll a[],ll ans[])
{
	if(n==1){ans[0]=ksm(a[0],mod-2);return;}
	solve((n+1)/2,a,ans);
	len=1;
	while(len<2*n) len*=2;
	for(int i=0;i<n;i++) a1[i]=a[i];
	for(int i=n;i<len;i++) a1[i]=0;
	change(a1),change(ans);
	ntt(a1,1),ntt(ans,1);
	for(int i=0;i<len;i++) ans[i]=ans[i]*(2-ans[i]*a1[i]%mod+mod)%mod;
	change(ans),ntt(ans,-1);
	for(int i=n;i<len;i++) ans[i]=0;
}
int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++) scanf("%lld",&a[i]),a[i]%=mod;
	solve(n,a,ans);
	for(int i=0;i<n;i++) printf("%lld ",ans[i]);
}