C++ アルゴリズムとデータ構造のライブラリ
#include "library/polynomial/fft/fast_walsh_hadamard_transform.hpp"$O(NlogN)$
fast_walsh_hadamard_transform(f, false);
#pragma once
template <typename T>
void fast_walsh_hadamard_transform(vector<T> &f, bool inv = false) {
const int n = (int)f.size();
assert((n & (n - 1)) == 0);
for (int i = 1; i < n; i <<= 1) {
for (int j = 0; j < n; j += i << 1) {
for (int k = 0; k < i; k++) {
T s = f[j + k], t = f[j + k + i];
f[j + k] = s + t;
f[j + k + i] = s - t;
}
}
}
if (inv) {
T inv_n = T(1) / n;
for (auto &x : f) x *= inv_n;
}
}#line 2 "library/polynomial/fft/fast_walsh_hadamard_transform.hpp"
template <typename T>
void fast_walsh_hadamard_transform(vector<T> &f, bool inv = false) {
const int n = (int)f.size();
assert((n & (n - 1)) == 0);
for (int i = 1; i < n; i <<= 1) {
for (int j = 0; j < n; j += i << 1) {
for (int k = 0; k < i; k++) {
T s = f[j + k], t = f[j + k + i];
f[j + k] = s + t;
f[j + k + i] = s - t;
}
}
}
if (inv) {
T inv_n = T(1) / n;
for (auto &x : f) x *= inv_n;
}
}