loj 3120 黑科技做法

牛逼的杜ls在CF上面发了一个生成函数黑科技。

设$k=n-2m$,并且略去若干种边界情况。

将答案写成生成函数形式，形如:$\frac{n!}{2^D}[x^n]\sum _{i=0}^k\left(\genfrac{}{}{0ex}{}{D}{i}\right)(e^x-e^{-x}{)}^i(e^x+e^{-x}{)}^{D-i}$。

设$Q(x)=\sum _{i=0}^k\left(\genfrac{}{}{0ex}{}{D}{i}\right)(x-1{)}^i(x+1{)}^{D-i}$。则答案等于$\frac{n!}{2^D}[x^n]Q(e^{2x})e^{-Dx}$。考虑如何快速求解$Q(x)$。

构造如下方程:$P(x)=DQ(x)-x{Q}^{\prime }(x)$，并且将右侧展开。

考虑$x{Q}^{\prime }(x)$的实际含义，不难发现其为任选一个$x\pm 1$替换为$x$的结果。假设我们替换掉的是第$i$个单项式。

同时我们将$DQ(x)$与$D$种替换掉的单项式一一对应。

此时若剩余单项式中$x-1$的个数不为k，此时被替换的单项式可以为$x-1$，也可以为$x-1$。做完差后为$-1$与$1$，对答案贡献为0可以忽略不计。

剩余单项式中$x-1$的个数为k，此时被替换的单项式仅能为$x+1$。会对差产生1的贡献。

因此$P(x)=D\left(\genfrac{}{}{0ex}{}{D-1}{k}\right)(x-1{)}^k(x+1{)}^{D-k-1}$。

考虑如何求$P(x)$。一般的设$P(x)=(x-1{)}^a(x+1{)}^b$。

考虑$P(x)$求导后与$P(x)$的关系式子，不难得到$P^{\prime }(x)=\frac{aP(x)}{x-1}+\frac{bP(x)}{x+1}$。

根据关系式子我们可以从高到底按位确定$P(x)$的值，因此即可按位确定$Q(x)$的值。

复杂度瓶颈在预处理$1^n,2^n,\cdots ,D^n$上，由于其为积性函数可以使用线性筛求解。

时间复杂度$O(m(1+{\log }_{m}n))$。

代码链接。