math/transforms/fftMul.cpp


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

vector<cplx> mul(vector<cplx>& a, vector<cplx>& b) {
	vector<cplx> c(a.size()), d(a.size());
	for (int i = 0; i < b.size(); i++) {
		c[i] = {real(a[i]), real(b[i])};
	}
	c = fft(c);
	for (int i = 0; i < b.size(); i++) {
		int j = (a.size() - i) % a.size();
		cplx x = (c[i] + conj(c[j])) / cplx{2, 0}; //fft(a)[i];
		cplx y = (c[i] - conj(c[j])) / cplx{0, 2}; //fft(b)[i];
		d[i] = x * y;
	}		
	return fft(d, true);
}