营业日志 2020.8.25 单位根与等价类

一年前的 BJ 集训里出现了这样一道题，其中我留下了这样一个疑惑：

计算 ${\prod }_{i=0}^{n-1}(1+x^i)\mathrm{m}\mathrm{o}\mathrm{d}(x^n-1)$ 的系数。

当然这个问题的第一个 observation 是非常重要的，即对于 $\gcd (n,k)=g$ 有相同 $g$ 的 $k$，各 $k$ 次项系数是相等的。因此我们可以认为这个问题只要求 $d(n)$ 个 output。

这是因为我们考虑 $x\to x^k$ 的映射，当 $\gcd (n,k)=1$ 的时候对下标是一个置换，在这个置换下乘积式显然不变，而取遍所有互质的 $k$ 的时候每一个 $\gcd$ 相同的下标组则构成一个轨道。

那道题里有个看起来像是 $\Theta (n^2)$ 的瓶颈，所以这部分就 $\Theta (n^2)$ 过了，但是今天这个部分重新拿到我面前的时候，总算经过重新思考，想到了一些有趣的东西。

我们考虑其 DFT 数组，首先不难观察到 DFT 数组也和第一个 observation 是相同的，这是可以考虑根的置换符合这一性质。

因此 $\mathbf{D}\mathbf{F}\mathbf{T}[a{]}_i$ 我们设 $\gcd (i,n)=k$，那么这个 DFT 值其实就是

$$\begin{array}{rl}& (1+{\omega }_n^k)(1+{\omega }_n^{2k})\cdots (1+{\omega }_n^{nk})\\ & ={\left[(1+{\omega }_n^k)(1+{\omega }_n^{2k})\cdots (1+{\omega }_n^{\frac{n}{k}k})\right]}^k\\ & ={\left[(1+{\omega }_{n/k})(1+{\omega }_{n/k}^2)\cdots (1+{\omega }_{n/k}^{n/k})\right]}^k\end{array}$$

所以我们还是主要要解决 $(1+{\omega }_n)(1+{\omega }_n^2)\cdots (1+{\omega }_n^n)$ 是多少，其实我们不妨做变形：

$$P(x)=(1-{\omega }_{n}x)(1-{\omega }_n^{2}x)\cdots (1-{\omega }_n^{n}x)=1-x^n$$

由根系关系，这是显然的，而我们只需带入 $x=-1$，便发现上式的乘积即为 $(1-(-1{)}^n)=2[n\mathrm{m}\mathrm{o}\mathrm{d}2]$

因此对于 $\gcd (i,n)=k$ 的所有 $i$，我们有

$$\mathbf{D}\mathbf{F}\mathbf{T}[a{]}_i=(2[n/k\mathrm{m}\mathrm{o}\mathrm{d}2]{)}^k$$

再做 IDFT 的时候，我们需要考虑相同取值位置的单位根值之和，即

$$\begin{array}{rl}{W}_j(n/k)& =\sum _{\gcd (i,n)=k}{\omega }_n^{ij}\\ & =\sum _{\gcd (i/k,n/k)=1}{\omega }_{n/k}^{ij/k}\\ W_j(n)& =\sum _{\gcd (i,n)=1}{\omega }_n^{ij}\\ \sum _{k|n}{W}_j(k)& =\sum _{0\le i<n}{\omega }_n^{ij}\\ & =n[n|j]\end{array}$$

于是，我们完成了 $\Theta (\mathrm{poly}n)$ 到 $\Theta (\mathrm{poly}d(n))$ 的跨越，剩下的优化有兴趣者可以继续完成。