单位根反演练习题

loj6485 LJJ学二项式定理

首先枚举$i(\mathrm{m}\mathrm{o}\mathrm{d}4)$的余数$t$，然后转化成对于每一个$t$，求

$$\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^i[4\mid (i-t)]$$

单位根反演：

$$\begin{gathered} =\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^i\frac{1}{4}\sum _{j=0}^3{\omega }_4^{j(i-t)} \\ =\frac{1}{4}\sum _{j=0}^3{\omega }_4^{-jt}\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^i{w}_4^{ij} \end{gathered}$$

最后面的那一坨东西，由二项式定理知$(1+x{)}^n={\sum }_{i=0}^n(\genfrac{}{}{0ex}{}{n}{i})x^i$，所以：

$$=\frac{1}{4}\sum _{j=0}^3{\omega }_4^{-jt}(s{\omega }_4^j+1{)}^n$$

直接算就可以了。

Code

uoj450【集训队作业2018】复读机

考虑一个复读机复读次数的生成函数：

$$F(x)=\sum _{i=0}^{\infty }\frac{x^i}{i!}[d\mid i]$$

而我们要算的实际上是$n!\cdot F^k(x)[x^n]$。

对$F(x)$进行单位根反演：

$$\begin{gathered} F(x)=\frac{1}{d}\sum _{i=0}^{\infty }\frac{x^i}{i!}\sum _{j=0}^{d-1}{\omega }^{ij} \\ =\frac{1}{d}\sum _{j=0}^{d-1}\sum _{i=0}^{\infty }\frac{x^i{w}^{ij}}{i!} \\ =\frac{1}{d}\sum _{j=0}^{d-1}{e}^{{\omega }^{j}x} \end{gathered}$$

$\frac{1}{d}$是常数，我们先把它忽视掉，最后再在答案里乘上一个$\frac{1}{d^k}$。

考虑$F^k(x)$的组合意义，相当于对于这$k$个多项式，从每一个里面选一项相乘，所有的方案得到的乘积的和。枚举$e^{{\omega }^{j}x}$被选的次数$p_j$，得到
$$\begin{gathered} F^k(x)=\sum _{p_0+p_1\cdots p_{d-1}=k}k!\prod _{j=0}^{d-1}\frac{1}{p_j!}{e}^{p_j{\omega }_{j}x} \\ =\sum _{p_0+p_1\cdots p_{d-1}=k}{e}^{({\sum }_{j=0}^{d-1}{p}_j{w}_j)x}\cdot k!\prod _{j=0}^{d-1}\frac{1}{p_j!} \end{gathered}$$

而

$$F^k(x)[x^n]=\sum _{p_0+p_1\cdots p_{d-1}=k}\frac{({\sum }_{j=0}^{d-1}{p}_j{w}_j{)}^n}{n!}\cdot k!\prod _{j=0}^{d-1}\frac{1}{p_j!}$$

直接计算，复杂度$O(k^d\log n)$。

Code

bzoj3328 PYXFIB

设
$$\begin{gathered} A=\left[\begin{array}{cc}1& 0\\ 1& 1\end{array}\right] \\ B=\left[\begin{array}{c}1\\ 1\end{array}\right] \end{gathered}$$

则我们相当于要求出

$$(\sum _{i=0}^n(\genfrac{}{}{0ex}{}{n}{i})A^i[k\mid i])B$$

对前面的那个括号里面的东西单位根反演，得到

$$\begin{gathered} \frac{1}{k}\sum _{j=0}^{k-1}\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)A^i{\omega }^{ij} \\ =\frac{1}{k}\sum _{j=0}^{k-1}({\omega }^j\cdot A+E{)}^n \end{gathered}$$

其中$E$表示单位矩阵。

直接矩阵快速幂计算，时间复杂度为$O(2^3\cdot \log n\cdot k)$。

loj3058 「HNOI2019」白兔之舞

设矩阵$A$是满足$A_{i,j}=w(i,j)$的$n\times n$的矩阵。

走$m$步的方案数为$\left(\genfrac{}{}{0ex}{}{L}{m}\right)T_{x,y}$，其中$T=A^m$。

则对于每一个$t$，我们需要求出

$$\sum _{m=0}^L\left(\genfrac{}{}{0ex}{}{L}{m}\right)A^m[k\mid (t-m)]$$

单位根反演之后得到

$$\frac{1}{k}\sum _{j=0}^{k-1}{\omega }_k^{-tj}({\omega }_k^{j}A+E{)}^L$$

对于每一个$j$，分别算出$F_j=({\omega }_k^{j}A+E{)}^L$，设$f_j$为$F_j$的$x$行$y$列的元素，则$t$的答案就是

$$An{s}_t=\frac{1}{k}\sum _{j=0}^{k-1}{f}_j{\omega }_k^{-tj}$$

由于$-tj=\left(\genfrac{}{}{0ex}{}{t}{2}\right)+\left(\genfrac{}{}{0ex}{}{j}{2}\right)-\left(\genfrac{}{}{0ex}{}{t+j}{2}\right)$，所以

$$An{s}_t=\frac{1}{k}\cdot {\omega }_k^{\left(\genfrac{}{}{0ex}{}{t}{2}\right)}\sum _{j=0}^{k-1}(f_j\cdot {\omega }_k^{\left(\genfrac{}{}{0ex}{}{j}{2}\right)}){\omega }_k^{-\left(\genfrac{}{}{0ex}{}{i+j}{2}\right)}$$

这是一个卷积的形式，用MTT优化一下就可以了。

总时间复杂度$O(k{n}^3\log L+k\log k)$。