python3的fft_FFT乘法Python 3.4.3

我有FFT和IFFT函数。我知道A*B = IFFT(FFT(A)*FFT(B))

在哪里FFT(A)FFT(B)=[qw for q,w in zip(A,B)]

但当我输入：10 10=&gt；输出：[(0.5+0j)，(0.5+0j)]

我做错什么了？

这是我的代码：from cmath import exp,pi

def FFT(X):

n = len(X)

w = exp(-2*pi*1j/n)

if n > 1:

X = FFT(X[::2]) + FFT(X[1::2])

for k in range(n//2):

xk = X[k]

X[k] = xk + w**k*X[k+n//2]

X[k+n//2] = xk - w**k*X[k+n//2]

return X

def IFFT(X):

n = len(X)

w = exp(2*pi*1j/n)

if n > 1:

X = IFFT(X[::2]) + IFFT(X[1::2])

for k in range(n//2):

xk = X[k]

X[k] = (xk + w**k*X[k+n//2])/n

X[k+n//2] = ((xk - w**k*X[k+n//2]))/n

return X

s = input().split()

a1 = s[0]

b1 = s[1]

a = [int(x) for x in a1]

b = [int(x) for x in b1]

if (len(a)>len(b)):

n = len(a)

b.reverse()

while (len(b)

b.extend([0])

b.reverse()

else:

n = len(b)

a.reverse()

while(len(a)

a.extend([0])

a.reverse()

c = [q*w for q,w in zip(a,b)]

print (IFFT(c))