fft卡常研究

// 1. 注意加減乘除寫在struct裏 且 加上&
// 2. w預處理不一定好,反而麻煩,我們只要連續
// 3. reverse可以代替-w
// 4. 注意rev的預處理,這個還是有點用的

#include <bits/stdc++.h>

#define I register int
#define F(i, a, b) for (I i = a; i <= b; i ++)

typedef double db;
const db pi = acos(- 1);

using namespace std;

int n, nn, Len, rev[M];

struct Z {
	db x, y;
	Z operator * (Z &a) { return Z{x * a.x - y * a.y, x * a.y + y * a.x}; }
	Z operator + (Z &a) { return Z{x + a.x, y + a.y}; }
	Z operator - (Z &a) { return Z{x - a.x, y - a.y}; }
} A[M], B[M];

void Dft(Z *A, I sig) {
	F(i, 1, Len - 1)
		if (i < rev[i]) swap(A[i], A[rev[i]]);
	for (I m = 2; m <= Len; m <<= 1) {
		Z w = Z{cos(2 * pi / m), sin(2 * pi / m)};
		I half = m >> 1;
		for (I i = 0; i < Len; i += m) {
			Z o = Z{1, 0};
			for (I j = i; j < i + half; j ++, o = o * w) {
				Z u = A[j + half] * o;
				A[j + half] = A[j] - u;
				A[j] = A[j] + u;
			}
		}
	}
	if (sig == - 1) {
		reverse(A + 1, A + Len);
		F(i, 0, Len - 1)
			A[i].x = int(A[i].x / Len + 0.5);
	}
}

int main() {
// Do sth..
// nn is the length of A,B
	for (Len = 1, cnt = 0; Len <= nn * 2; Len <<= 1, cnt ++);

	F(i, 1, Len - 1)
		rev[i] = (rev[i >> 1] >> 1) ^ (i & 1) * (Len >> 1);
	Dft(A, 1), Dft(B, 1);
	F(i, 0, Len - 1) A[i] = A[i] * B[i];
	Dft(A, - 1);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章