概述
多项式求逆元是一个非常重要的知识点,许多多项式操作都需要用到该算法,包括多项式取模,除法,开跟,求ln,求exp,快速幂。用快速傅里叶变换和倍增法可以在$O(n log n)$的时间复杂度下求出一个$n$次多项式的逆元。
前置技能
快速数论变换(NTT),求一个数$x$在模$p$意义下的乘法逆元。
多项式的逆元
给定一个多项式$A(x)$,其次数为$deg_A$,若存在一个多项式$B(x)$,使其满足$deg_B≤deg_A$,且$A(x)\times B(x) \equiv 1 (mod\ x^n)$,则$B(x)$即为$A(x)$在模$x^n$意义下的的乘法逆元。
求多项式的逆元
我们不妨假设,$n=2^k,k∈N$。
若$n=1$,则$A(x)\times B(x) \equiv a_0\times b_0 \equiv 1 (mod\ x^1)$。其中$a_0$,$b_0$表示多项式$A$和多项式$B$的常数项。
若需要求出$b_0$,直接用费马小定理求出$a_0$的乘法逆元即可。
当$n>1$时:
我们假设在模$x^{\frac{n}{2}}$的意义下$A(x)$的逆元$B'(x)$我们已经求得。
依据定义,则有
$A(x)B'(x)\equiv 1 (mod\ x^{\frac{n}{2}})$ $(1)$
对$(1)$式进行移项得
$A(x)B'(x)-1\equiv 0 (mod\ x^{\frac{n}{2}})$ $(2)$
然后对$(2)$式等号两边平方,得
$A^2(x)B'^2(x)-2A(x)B'(x)+1\equiv 0(mod\ x^{n})$ $(3)$
将常数项移动到等式右侧,得
$A^2(x)B'^2(x)-2A(x)B'(x)\equiv -1(mod\ x^{n})$ $(4)$
将等式两边去相反数,得
$2A(x)B'(x)-A^2(x)B'^2(x)\equiv 1(mod\ x^{n})$ $(5)$
下面考虑回我们需要求的多项式$B(x)$,依据定义,其满足
$A(x)B(x)\equiv 1(mod\ x^{n}) $(6)$
将$(5)-(6)$并移项,得
$A(x)B(x)\equiv 2A(x)B'(x)-A^2(x)B