content/math/transforms/bitwiseTransforms.cpp


1
2
3
4
5
6
7
8
9
10
11
12

void bitwiseConv(vector<ll>& a, bool inv = false) {
	int n = ssize(a);
	for (int s = 1; s < n; s *= 2) {
		for (int i = 0; i < n; i += 2 * s) {
			for (int j = i; j < i + s; j++) {
				ll &u = a[j], &v = a[j + s];
				u = inv ? u - v : u + v; // AND
				//v = inv ? v - u : v + u; // OR
				//tie(u, v) = pair(u + v, u - v); // XOR
	}}}
	//if (inv) for (ll& x : a) x /= n; // XOR (careful with MOD)
}