多项式平移|连续点值平移

多项式平移

多项式平移是简单情况的多项式复合变换,给出 f(x)=\sum _ {i=0}^nf_ix^i 的系数和一个常数 c ,求 f(x+c) 的系数,即 f(x)\mapsto f(x+c)

分治法

f(x)=f_0(x)+x^{\left\lfloor n/2\right\rfloor}f_1(x)

那么

f(x+c)=f_0(x+c)+(x+c)^{\left\lfloor n/2\right\rfloor}f_1(x+c)

(x+c)^{\left\lfloor n/2\right\rfloor} 的系数为二项式系数,那么

T(n)=2T(n/2)+O(n\log n)=O(n\log^2 n)

其中 O(n\log n) 为多项式乘法的时间。

Taylor 公式法

f(x) c 处应用 Taylor 公式,有

f(x)=f(c)+\frac{f'(c)}{1!}(x-c)+\frac{f''(c)}{2!}(x-c)^2+\cdots +\frac{f^{(n)}(c)}{n!}(x-c)^n

那么

f(x+c)=f(c)+\frac{f'(c)}{1!}x+\frac{f''(c)}{2!}x^2+\cdots +\frac{f^{(n)}(c)}{n!}x^n

观察到对于 t\geq 0

\begin{aligned} t!\lbrack x^t\rbrack f(x+c)&=f^{(t)}(c)\\ &=\sum _ {i=t}^nf_ii!\frac{c^{i-t}}{(i-t)!}\\ &=\sum _ {i=0}^{n-t}f _ {i+t}(i+t)!\frac{c^i}{i!} \end{aligned}

\begin{aligned} A_0(x)&=\sum _ {i=0}^nf _ {n-i}(n-i)!x^i\\ B_0(x)&=\sum _ {i=0}^n\frac{c^i}{i!}x^i \end{aligned}

那么

\begin{aligned} \lbrack x^{n-t}\rbrack (A_0(x)B_0(x))&=\sum _ {i=0}^{n-t} (\lbrack x^{n-t-i}\rbrack A_0(x))(\lbrack x^i\rbrack B_0(x))\\ &=\sum _ {i=0}^{n-t}f _ {i+t}(i+t)!\frac{c^i}{i!}\\ &=t!\lbrack x^t\rbrack f(x+c) \end{aligned}

二项式定理法

考虑二项式定理 \displaystyle (a+b)^n=\sum _ {i=0}^n\binom{n}{i}a^ib^{n-i} 那么

\begin{aligned} f(x+c)&=\sum _ {i=0}^nf_i(x+c)^i\\ &=\sum _ {i=0}^nf_i\left(\sum _ {j=0}^i\binom{i}{j}x^jc^{i-j}\right)\\ &=\sum _ {i=0}^nf_ii!\left(\sum _ {j=0}^i\frac{x^j}{j!}\frac{c^{i-j}}{(i-j)!}\right)\\ &=\sum _ {i=0}^n\frac{x^i}{i!}\left(\sum _ {j=i}^{n}f_jj!\frac{c^{j-i}}{(j-i)!}\right) \end{aligned}

得到的结果与上述方法相同。

连续点值平移

例题 LOJ 166 拉格朗日插值 2

给出度数小于等于 n 的多项式 f 的连续点值 f(0),f(1),\dots ,f(n) ,在模 998244353 意义下计算 f(c),f(c+1),\dots ,f(c+n) ,其中 1\leq n\leq 10^5,n < m\leq 10^8

Lagrange 插值公式法

考虑 Lagrange 插值公式

\begin{aligned} f(x)&=\sum _ {0\leq i\leq n}f(i)\prod _ {0\leq j\leq n\,\land \,j\neq i}\frac{x-j}{i-j}\\ &=\sum _ {0\leq i\leq n}f(i)\frac{x!}{(x-n-1)!(x-i)}\frac{(-1)^{n-i}}{i!(n-i)!}\\ &=\frac{x!}{(x-n-1)!}\sum _ {0\leq i\leq n}\frac{f(i)}{(x-i)}\frac{(-1)^{n-i}}{i!(n-i)!} \end{aligned}

上式虽然是卷积形式但不能保证分母上 x-i\neq 0 ,所以下面仅考虑 c > n 的情况,其他情况(如系数在模素数意义下时须避免 B_0(x) 系数的分母出现零)可以分类讨论解决,令

\begin{aligned} A_0(x)&=\sum _ {0\leq i\leq n}\frac{f(i)(-1)^{n-i}}{i!(n-i)!}x^i\\ B_0(x)&=\sum _ {i\geq 0}\frac{1}{c-n+i}x^i \end{aligned}

那么对于 t\geq 0

\begin{aligned} \lbrack x^{n+t}\rbrack (A_0(x)B_0(x))&=\sum _ {i=0}^{n+t}(\lbrack x^i\rbrack A_0(x))(\lbrack x^{n+t-i}\rbrack B_0(x))\\ &=\sum _ {i=0}^{n}\frac{f(i)(-1)^{n-i}}{i!(n-i)!}\frac{1}{c+t-i}\\ &=\frac{(c+t-n-1)!}{(c+t)!}f(c+t) \end{aligned}

实现中取 B_0(x) 需要的部分截断可求出更多点值,且可利用循环卷积。

对问题稍加修改,假设对于某个 d 给出的点值为 f(d),f(d+k),\dots ,f(d+nk) ,我们可以计算 f(c+d),f(c+d+k),\dots ,f(c+d+nk) ,视作平移 g(x)=f(d+kx) 的点值 g(0),g(1),\dots ,g(n) g(c/k),g(c/k+1),\dots ,g(c/k+n)

Lagrange 插值公式也给出了通过维护一些前后缀积的线性计算单个点值的方法。

应用

同一行第一类无符号 Stirling 数

例题 P5408 第一类斯特林数·行

在模素数 167772161 意义下求 \displaystyle {n\brack 0},{n\brack 1},\dots ,{n\brack n} ,其中 1\leq n< 262144

考虑

x^{\overline{n}}=\sum _ {i=0}^n{n\brack i}x^i,\quad n\geq 0

其中 x^{\overline{n}}=x\cdot (x+1)\cdots (x+n-1) 为上升阶乘幂,令 f_n(x)=x^{\overline{n}} 那么

f_{2n}(x)=x^{\overline{n}}\cdot (x+n)^{\overline{n}}=f_n(x)f_n(x+n)

通过多项式平移可在 O(n\log n) 求出 f_n(x+n) ,问题被缩小为原先的一半即求出 f_n(x) 的系数,那么

T(n)=T(n/2)+O(n\log n)=O(n\log n)

模素数意义下阶乘

例题 P5282 【模板】快速阶乘算法

n!\bmod p ,其中 p 为素数且 1\leq n< p\leq 2^{31}-1

v=\lfloor\sqrt{n}\rfloor g(x)=\prod _ {i=1}^v(x+i) 那么

n!\equiv \left(\prod _ {i=0}^{v-1}g(iv)\right)\cdot \prod _ {i=v^2+1}^n i\pmod{p}

其中 \prod _ {i=v^2+1}^n i 可在 O(\sqrt{n}) 时间计算,我们希望可以快速计算上式的前半部分。

多项式多点求值

g(x) 系数的计算可用上述多项式平移算法在 O(n\log n) 时间得到,但多点求值计算 g(0),g(v),g(2v),\dots ,g(v^2-v) 需要 O(\sqrt{n}\log ^2n) 时间。

连续点值平移

g_d(x)=\prod _ {i=1}^d(x+i) ,我们可以用 d+1 个点值 g_d(0),g_d(v),\dots ,g_d(dv) 唯一确定这个度数为 d 的多项式,又

g _ {2d}(x)=g_d(x)g_d(x+d)

所以只需 2d+1 个点值可以唯一确定 g _ {2d}(x) ,那么使用连续点值平移计算 g_d((d+1)v),g_d((d+2)v),\dots ,g_d(2dv) (即平移 h(x)=g_d(vx) 的点值 h(0),h(1),\dots ,h(d) h(d+1),h(d+2),\dots ,h(2d) )和 g_d(d),g_d(v+d),\dots ,g_d(2dv+d) (即平移 h(x)=g_d(vx) 的点值 h(0),h(1),\dots ,h(d) h(d/v),h(d/v+1),h(d/v+2),\dots ,h(d/v+2d) )后将这两者的对应点值相乘即得 g _ {2d}(0),g _ {2d}(v),\dots ,g _ {2d}(2dv)

g_d(0),g_d(v),\dots ,g_d(dv) 计算 g _ {d+1}(0),g _ {d+1}(v),\dots ,g _ {d+1}(dv),g _ {d+1}((d+1)v) 考虑

g _ {d+1}(x)=(x+d+1)\cdot g_d(x)

额外增加的一个点值使用线性时间的算法即可。那么在开始时维护 g_1(0)=1,g_1(v)=v+1 后使用连续点值平移来倍增地维护这些点值,有

T(n)=T(n/2)+O(\sqrt n\log n)=O(\sqrt n\log n)

模素数意义下二项式系数前缀和

例题 LOJ 6386 组合数前缀和

\displaystyle \sum _ {i=0}^m\binom{n}{i}\bmod 998244353 ,其中 0\leq m\leq n\leq 9\times 10^8

考虑使用矩阵描述 n!=n\cdot (n-1)! 这一步递推,我们有

\begin{bmatrix} n! \end{bmatrix} =\left( \prod _ {i=0}^{n-1} \begin{bmatrix}i+1\end{bmatrix} \right) \begin{bmatrix} 1 \end{bmatrix}

类似的可以将二项式系数前缀和的递推描述为

\begin{bmatrix} \binom{n}{m+1}\\ \sum _ {i=0}^m\binom{n}{i} \end{bmatrix}= \begin{bmatrix} (n-m)/(m+1)&0\\ 1&1 \end{bmatrix} \begin{bmatrix} \binom{n}{m}\\ \sum _ {i=0}^{m-1}\binom{n}{i} \end{bmatrix}

注意矩阵乘法的顺序,那么

\begin{aligned} \begin{bmatrix} \binom{n}{m+1}\\ \sum _ {i=0}^m\binom{n}{i} \end{bmatrix} &=\left( \prod _ {i=0}^{m} \begin{bmatrix} (n-i)/(i+1)&0\\1&1 \end{bmatrix} \right) \begin{bmatrix} 1\\0 \end{bmatrix}\\ &= \frac{1}{(m+1)!} \left( \prod _ {i=0}^{m} \begin{bmatrix} n-i&0\\i+1&i+1 \end{bmatrix} \right) \begin{bmatrix} 1\\0 \end{bmatrix} \end{aligned}

v=\lfloor\sqrt{m}\rfloor ,考虑维护矩阵

\begin{aligned} M _ d(x)&= \prod _ {i=1}^d \begin{bmatrix} -x+n+1-i&0\\ x+i&x+i \end{bmatrix}\\ &= \begin{bmatrix} f_d(x)&0\\ g_d(x)&h_d(x) \end{bmatrix} \end{aligned}

的点值 M _ d(0),M _ d(v),\dots ,M_d(dv) f_d(0),f_d(v),\dots ,f_d(dv) h_d(0),\dots ,h_d(dv) g_d(0),\dots ,g_d(dv) ,又

\begin{aligned} M _ {2d}(x)&= \prod _ {i=1}^{2d} \begin{bmatrix} -x+n+1-i&0\\ x+i&x+i \end{bmatrix}\\ &= \left( \prod _ {i=1}^d \begin{bmatrix} -x-d+n+1-i&0\\ x+d+i&x+d+i \end{bmatrix} \right) \left( \prod _ {i=1}^d \begin{bmatrix} -x+n+1-i&0\\ x+i&x+i \end{bmatrix} \right) \\ &= \begin{bmatrix} f_d(x+d)&0\\ g_d(x+d)&h_d(x+d) \end{bmatrix} \begin{bmatrix} f_d(x)&0\\ g_d(x)&h_d(x) \end{bmatrix} \\ &= \begin{bmatrix} f_d(x+d)f_d(x)&0\\ g_d(x+d)f_d(x)+h_d(x+d)g_d(x)&h_d(x+d)h_d(x) \end{bmatrix} \end{aligned}

且矩阵右下角元素恰为我们在阶乘算法中所维护的,那么

\begin{aligned} \prod _ {i=0}^{m} \begin{bmatrix} n-i&0\\i+1&i+1 \end{bmatrix}= \left( \prod _ {i=(k+1)v}^m \begin{bmatrix} n-i&0\\ i+1&i+1 \end{bmatrix} \right) \begin{bmatrix} f_v(kv)&0\\ g_v(kv)&h_v(kv) \end{bmatrix} \cdots \begin{bmatrix} f_v(0)&0\\ g_v(0)&h_v(0) \end{bmatrix} \end{aligned}

可在 O(\sqrt m\log m) 时间完成计算。

模素数意义下调和数

例题 P5702 调和级数求和

\sum _ {i=1}^ni^{-1}\bmod p ,其中 p 为素数且 1\leq n< p< 2^{30}

H_n=\sum _ {k=1}^nk^{-1} ,一步递推为

\begin{bmatrix} (n+1)!\\(n+1)!H _ {n+1} \end{bmatrix}= \begin{bmatrix} n+1&0\\1&n+1 \end{bmatrix} \begin{bmatrix} n!\\n!H_n \end{bmatrix}

那么

\begin{bmatrix} {n+1\brack 1}\\{n+1\brack 2} \end{bmatrix}= \begin{bmatrix} n!\\n!H_n \end{bmatrix}= \left( \prod _ {i=0}^{n-1} \begin{bmatrix} i+1&0\\1&i+1 \end{bmatrix} \right) \begin{bmatrix} 1\\0 \end{bmatrix}

在这里 \displaystyle {n+1\brack 1} \displaystyle {n+1\brack 2} 为第一类无符号 Stirling 数。维护点值矩阵的方法同上。

参考文献

  • Alin Bostan, Pierrick Gaudry, and Eric Schost. Linear recurrences with polynomial coefficients and application to integer factorization and Cartier-Manin operator.
  • Min_25 的博客
  • ZZQ 的博客 - 阶乘模大质数

评论