关于一类位运算多项式相乘的做法

题目大意

给定两个下标均为$[0..2^n - 1]$的多项式$A,B$，数组$C = A \times B$，满足
$C_i = \sum_{(j~op~k) = i} A_j * B_k$

其中$op$可以为三种位运算符号$or,and,xor$。

题解

这种问题有一个经典的做法，Fast Walsh-Hadamard Transform，但很难记忆，下面是种比较好推的做法。三种符号分类讨论。

Or

设当前$A,B$长度为$2L$，$A = a + b*x^L,B = c + d * x^L$，其实就是拆成两半。而且由于操作符是$Or$，所以可以直接通过$*x^L$实现左移$L$位。

那么$A * B = (a + bx^L) * (c + dx^L) = ac + x^L(bc + ad) + x^{L} * x^L * bd$

因为是$Or$，所以$x^L * x^L = x^L$，因此$A * B = ac + x^L(bc + ad + bd)$

那么我们可以先求出$(a + b) * (c + d)$以及$a*c$，然后用$O(L)$的时间就可以求出$A * B$了。

时间复杂度$T(L) = 2 * T(\frac{L}{2}) + O(L) = O(L * \log L)$

Xor

设当前$A,B$长度为$2L$，$A = a + b*x^L,B = c + d * x^L$，其实就是拆成两半。而且由于操作符是$Xor$，所以可以直接通过$*x^L$实现左移$L$位。

那么$A * B = (a + bx^L) * (c + dx^L) = ac + x^L(bc + ad) + x^{L} * x^L * bd$

因为是$Xor$，所以$x^L * x^L = x^0 = 1$，因此$A * B = ac + x^L(bc + ad) + bd$

那么设$X = (a + b) * (c + d),Y = (a - b) * (c - d)$，可以惊奇的发现$bc + ad = \frac{X - Y}{2},ac + bd = \frac{X + Y}{2}$。于是一样用$O(L)$的时间就可以求出了。

时间复杂度$T(L) = 2 * T(\frac{L}{2}) + O(L) = O(L * \log L)$

And

一样把$A,B$分为两半，但现在不能用$x^L$来左移了，因为是$and$。

但是可以发现的是，$A * B$的左边只能由$b*d$做出贡献，而且右边的贡献就是$bc+ad+ac$。

那么我们可以算出$(a + b) * (c + d),b * d$,然后拼接一下就好了。

时间复杂度$T(L) = 2 * T(\frac{L}{2}) + O(L) = O(L * \log L)$