快速沃尔什变换 [学习笔记]

快速沃尔什变换

概述

用来解决一类与位运算有关卷积问题：

[C_i = sum_{j oplus k = i}A_j * B_K ]

---

过程

具体看picks的博客和这两篇blog吧：1 2

基础思想和fft类似，我们正变换求出一个类似点值表示的东西，然后用它直接乘，然后逆变换。

fft我们对下标奇偶分治。这里求变换我们按位分治。

与和或根据位运算的性质很好想，可以自己推一推异或记住就行了

二进制运算的性质很强啊，重点在于没有进位！所以说类似快速幂可以直接对“点值表示”做.

---

与

[A=(A_0,A_1) \ fwt(A) = (fwt(A_0)+fwt(A_1), fwt(A_1)) \ ifwt(A) = (ifwt(A_0)-ifwt(A_1), ifwt(A_1)) \ ]

或

[A=(A_0,A_1) \ fwt(A) = (fwt(A_0), fwt(A_1) + fwt(A_0)) \ ifwt(A) = (ifwt(A_0), ifwt(A_1)- ifwt(A_0)) \ ]

异或

[A=(A_0,A_1) \ fwt(A) = (fwt(A_0)+fwt(A_1), fwt(A_0)-fwt(A_1)) \ ifwt(A) = (ifwt(frac{A_0+A_1}{2}), ifwt(frac{A_0-A_1}{2})) \ ]

---

代码：

4589: Hard Nim

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = (1<<17)+5, P = 1e9+7, inv2 = (P+1)/2;

bool notp[N]; int p[N/10];
void sieve(int n) {
	for(int i=2; i<=n; i++) {
		if(!notp[i]) p[++p[0]] = i;
		for(int j=1; j<=p[0] && i*p[j]<=n; j++) {
			notp[i*p[j]] = 1;
			if(i%p[j] == 0) break;
		}
	}
}

void fwt(int *a, int n, int flag) {
	for(int l=2; l<=n; l<<=1) {
		int m = l>>1;
		for(int *p = a; p != a+n; p += l) 
			for(int k=0; k<m; k++) {
				int x = p[k], y = p[k+m];
				if(flag == 1) p[k] = (x + y) %P, p[k+m] = (x - y + P) %P;
				else p[k] = (ll) (x + y) * inv2 %P, p[k+m] = (ll) (x - y + P) * inv2 %P;
			}
	}
}

int Pow(ll a, int b) {
	ll ans = 1;
	for(; b; b>>=1, a=a*a%P)
		if(b&1) ans=ans*a%P;
	return ans;
}

int n, m, a[N];
int main() {
	freopen("in", "r", stdin);
	sieve(N-1);
	while(scanf("%d %d", &n,&m) != EOF) {
		memset(a, 0, sizeof(a));
		for(int i=1; i<=p[0] && p[i]<=m; i++) a[p[i]] = 1;
		int len = 1; while(len <= m) len<<=1;
		fwt(a, len, 1);
		for(int i=0; i<len; i++) a[i] = Pow(a[i], n);
		fwt(a, len, -1);
		printf("%d
", a[0]);
	}
}