blob: 9e40c74206fe9d4708d8e44d78ccf3ba37bdbd1c (
plain)
1
2
3
4
5
6
7
8
|
void fft(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];
tie(u, v) = inv ? pair(u - v, v) : pair(u + v, v);
}}}}
|