一种多项式多点求值算法

2020 年 05 月 18 日发布.

Problem

给定一个 $m-1$ 次多项式 $f(x)$ 以及 $n$ 个点 $\alpha_0,\dots,\alpha_{n-1}$,求 $f(\alpha_0),f(\alpha_1),\dots,f(\alpha_{n-1})$。

参考:《转置原理的简单介绍》rushcheyo, negiizhao, Created_Equal

参考:EI's blog

参考:Tellegen’s Principle into Practice

Algorithm

大家可能都知道可以通过多项式取模来做多点求值(i.e.$f(\alpha)=f(x)\bmod(x-\alpha)$)。

我们考虑这个多点求值到底是个什么东西:

对于一个 $m-1$ 次多项式 $f(x)=\sum_{i=0}^{n-1}f_ix^i$,我们将其看做一个 $m$ 维列向量 $f=(f_0,f_1,\dots,f_{m-1})^T$。($-^T$ 表示转置)。那么我们求的是:对于每个 $i\in[0,n)$ 求 $\sum_{j=0}^{n-1}f_j\alpha_i^j$。换句话说我们要求

$$ \begin{pmatrix} 1&\alpha_0&\alpha_0^2&\cdots&\alpha_0^{m-1}\\ 1&\alpha_1&\alpha_1^2&\cdots&\alpha_1^{m-1}\\ 1&\alpha_2&\alpha_2^2&\cdots&\alpha_2^{m-1}\\ \vdots&\vdots&\vdots&\ddots&\vdots\\ 1&\alpha_{n-1}&\alpha_{n-1}^2&\cdots&\alpha_{n-1}^{m-1}\\ \end{pmatrix}f $$

固定 $\alpha_0,\dots\alpha_{n-1}$,左边其实就是一个范德蒙矩阵(Vandermonde Matrix),记它为 ${\bf V}$。

所以我们要做的其实就是要求:将一个向量左乘一个范德蒙矩阵。

这个东西不太好搞;但是如果考虑左乘它的转置( $u\mapsto {\bf V}^Tu$),那么这个我们是知道如何去做的:把式子展开,我们要求的是:对每个 $i\in[0,m)$,求出 $\sum_{j=0}^{n-1}u_j\alpha_j^i$。换句话说,我们要求 $\sum_{j=0}^{n-1}\frac{u_j}{1-\alpha_jx}$ 的前 $m$ 项。这个东西我们很会做:这就是分治 FFT 上去。

那么我们已知了如何从 $u$ 去求 ${\bf V}^Tu$。这对我们的目标(从 $f$ 求出 ${\bf V}f$)有什么帮助?

Tellegen's Principle

我们发现 $u\mapsto {\bf V}^Tu$ 这个计算是线性的(假设 $n,\alpha_0,\dots,\alpha_{n-1}$ 给定,把 $u_0\dots u_{n-1}$ 看作输入的话,可以发现整个算法中仅有加减以及乘除常数)。这也是很自然的,因为要计算的东西本来就正好是 $u$ 左乘一个矩阵。

既然这样,我们把 $u\mapsto {\bf V}^Tu$ 的整个程序展开,就一定会得到这样的一个过程:每一步都是如下三种操作的一种:

注意这里我们一直认为 $n,\alpha_0,\dots,\alpha_{n-1}$ 是固定的,即不把它们看做输入。

这里其实我们就是将 ${\bf V}^T$ 分解成了若干个初等行变换矩阵的乘积 ${\bf V}^T=E_kE_{k-1}\dots E_1$($k$ 表示把整个算法展开之后会有多少步)。因此我们就有 ${\bf V}=E_1^TE_2^T\dots E_k^T$;换句话说,我们只需要把上面每一种操作也“转置”过来,并把整个过程“反过来”,那么我们就得到了一个 $f\mapsto {\bf V}f$ 的算法;这里,第一种操作转置过来就会变成 ${\bf y}\mathrel{+}=c{\bf x}$;后两种操作转置过来是不变的。

这个操作(通过 $u\mapsto{\bf A}u$ 的算法来得到 $f\mapsto{\bf A}^Tf$ 的算法)被称作“转置原理”,或“特勒根原理”(Tellegen's Principle)。

举个例子:如果我们的输入是 $(f_0,f_1,f_2)$,输出是 $(f_0,f_0+f_1,f_0+f_1+f_2)$,那么有这样一个算法:$f_1\mathrel{+}=f_0;\;f_2\mathrel{+}=f_1$。转置过来之后的输出是 $(f_0+f_1+f_2,f_1+f_2,f_2)$,对应的算法是 $f_1\mathrel{+}=f_2;\;f_0\mathrel{+}=f_1$。

这样的话,我们至少可以“劝自己相信存在某个和原来一样复杂度一样常数的算法”,毕竟我们会做同样多次的加减法呢、同样多次的乘法。

实现

但是一个问题是,你显然不会蠢到在程序里面手动实现一个转置原理(比如,把你用到的加减乘除操作全都丢到一个队列里然后反过来做,虽然这是完全可行的)。当然你可以手动把程序反过来写,但是比较好的方法是手动分析这里用到的各种算法的”转置“是什么样的。

多项式乘法

首先肯定要考虑多项式乘法。固定一个 $n$ 次多项式 $g$,然后把一个 $m$ 次多项式 $f$ 看做输入,那么我们应该得到一个 $n+m$ 次多项式 $ans$。这个线性变换就是对每对 $i,j$,把 $g_i$ 倍的 $f_j$ 加到 $ans_{i+j}$ 上。因此它的转置应该是:把一个 $n+m$ 次多项式 $f$ 作为输入,得到一个 $n$ 次多项式 $ans$;然后对每对 $i,j$,把 $g_i$ 倍的 $f_{i+j}$ 加到 $ans_j$ 上(这个过程其实很简单,你只需要记住把“输入”到“输出”的系数反过来就好了)。因此我们得到的式子就是 $ans_j=\sum_ig_if_{i+j}$,其中 $ans$ 的次数不低于 $f,g$ 的次数之差(“不低于”是因为我们可能会把 $f$ 看做超过 $n+m$ 次多项式,因为我们正着 FFT 的时候会算 $n+m$ 次但只保留前若干项)。

这个过程可以通过把 $g$ 反过来 FFT 一次实现,我们把这样得到的结果 $ans$ 写作 $g\times^T_nf$(上标 $T$ 表示转置,而下标标出了结果需要保留多少次)。

多点求值

考虑我们计算 ${\bf V}^Tf$ 的算法:要计算 $\sum_{i=0}^{n-1}\frac{u_i}{1-\alpha_ix}$,记 $g(x)=\prod_{i=0}^{n-1}(1-\alpha_ix)$,然后我们计算 $\sum_{i=0}^{n-1}f_i\frac{g(x)}{1-\alpha_ix}$,最后再把它乘上 $g^{-1}(x)$。首先我们先利用分治乘法来计算 $g(x)$。由于这部分并不依赖于输入,所以它不属于转置的范畴。(再强调一次我们只把 $u_0,\dots,u_{n-1}$ 看作输入)

对于 $\sum_{i=0}^{n-1}u_i\frac{p(x)}{1-\alpha_ix}$,我们采用分治:

若这一步得到的答案记为 $h$,那么最终的答案(${\bf V}^Tu$)就是 $g^{-1}\times h$。

将这个算法转置过来,不要忘记要把操作的顺序也反过来:

首先仍然是分治计算 $g$,这一步是不变的;然后对于输入的 $f$,我们计算 $h=g^{-1}\times^T_nf$(注意,考虑到前面我们的 $h$ 有 $n$ 位,所以这里的 $h$ 也应该保留 $n$ 位。但是前面的 $g^{-1}$ 要计算 $m$ 位,所以这里的 $g^{-1}$ 也应该是计算完整的 $m$ 位)。

接下来把前面所说的分治转置过来:

这样我们就可以得到 ${\bf V}f$,即多点插值结果。

此外,在计算的时候不难发现 $\times^T$ 的操作所需要的 FFT 长度是和原来相同的(因为我们反过来卷积后只需要后面一半,所以 FFT 短一些,循环卷积溢出到前面那一半我们也用不到),所以这样的常数和前面的是相同的;我们只需要一次多项式求逆,别的都只是普通的fft。这相比于原来(每一步都要做一个多项式取模)快了不少。

Code

咕咕咕。