blob: 87bae0b1c129dc914f3278f7103df4d797f5828e (
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];
u = inv ? u - v : u + v;
}}}}
|