多项式复合逆和拉格朗日反演

引入

设有多项式 $F(x),G(x)$，满足常数项为 $0$ 且一次项不为 $0$

如果 $G(F(x))=x$ ，则有 $F(G(x))=x$ 。此时，称 $F$ 和 $G$ 互为复合逆。

拉格朗日反演及其扩展用于求一类 $G(F(x))=H(x)$ 的问题，其中 $G(x),H(x)$ 为已知多项式函数，$F(x)$ 为待求函数。但其只能在 $O(n\log n)$ 的时间内求得 $F(x)$ 的某一项系数。

拉格朗日反演

若 $G(F(x))=x$ ，则有
$$\begin{gathered} [x^n]F(x)=\frac{1}{n}[x^{-1}]\frac{1}{G^n(x)} \\ [x^n]H(F(x))=\frac{1}{n}[x^{-1}]H^{\prime }(x)\frac{1}{G^n(x)} \end{gathered}$$
这个 $[x^{-1}]$ 是什么东西，我学的形式幂级数没有 $x^{-1}$ 项啊。

我们把右边的式子乘以 $x^n$ ，系数取 $[x^{n-1}]$ ，那么
$$\begin{gathered} [x^n]F(x)=\frac{1}{n}[x^{n-1}](\frac{x}{G(x)}{)}^n \\ [x^n]H(F(x))=\frac{1}{n}[x^{n-1}]H^{\prime }(x)(\frac{x}{G^n(x)}{)}^n \end{gathered}$$

证明

$$\begin{array}{rl}G(F(x))& =x\\ \sum _{i=1}^{\infty }{a}_i{F}^i& =x\end{array}$$

两边同时对 $x$ 求导
$$\sum _{i=1}^{\infty }i{a}_i{F}^{i-1}{F}^{\prime }=1$$
两边同时除以 $F^n$ ，取 $x^{-1}$ 的系数
$$[x^{-1}]\sum _{i=1}^{\infty }i{a}_i{F}^{i-n-1}{F}^{\prime }=[x^{-1}]\frac{1}{F^n}$$
观察左边的式子

$i\ne n$ 时，$F^{i-n-1}{F}^{\prime }=\frac{1}{i-n}(F^{i-n}{)}^{\prime }$ 。我们知道一个任意一个多项式求导 $x^{-1}$ 的系数都为 $0$ 。所以 $[x^{-1}]i{a}_i{F}^{i-n-1}{F}^{\prime }=0$

$i=n$ 时，$F^{-1}{F}^{\prime }=\frac{a_1+2a_{2}x+3a_3{x}^2+...}{a_{1}x+2a_2{x}^2+3a_3{x}^3+...}=\frac{a_1+2a_{2}x+...}{a_{1}x}\cdot \frac{1}{1+(\frac{a_2}{a_1}x+\frac{a_3}{a_2}x+...)}$ 。

左边的分母 $1+(\frac{a_2}{a_1}x+\frac{a_3}{a_2}x+...)$ 是可逆的，且其逆的常数项的为 $1$ 。又因为 $[x^{-1}]\frac{a_1+2a_{2}x+...}{a_{1}x}=1$ ，所以 $[x^{-1}]F^{-1}{F}^{\prime }=1$ 。

所以原式可以化成
$$\begin{array}{rl}n{a}_n& =[x^{-1}]\frac{1}{F^n}\\ a_n& =\frac{1}{n}[x^{-1}]\frac{1}{F^n}\end{array}$$
发现 $F$ 没有逆，而且形式幂级数也没有 $x^{-1}$ 。则把给右边乘以 $x^n$ ，再取 $x^{n-1}$ 的系数就是
$$\begin{gathered} a_n=\frac{1}{n}[x^{n-1}](\frac{x}{F}{)}^n \\ [x^n]G=\frac{1}{n}[x^{n-1}](\frac{x}{F}{)}^n \end{gathered}$$
得证。

例题

【BZOJ3684】大朋友和多叉树