haleyk100198's blog

Need help for optimizing FFT runtime
By haleyk100198, history, 8 years ago, In English

I was practising FFT with some problems, but I found out that my implementation of FFT seems to take longer than what it takes to fit into the TL.

For the problem 528D. Fuzzy Search (I understand that the problem also has a bitmask solution, but I'd like to practise FFT), I implemented the FFT algorithm (code here) for the string comparison, yet it takes more than 3 seconds to compute 12 FFT(and inverse) of a vector with a size of (2^18).

I wonder how I can improve my implementation's runtime?

  • Vote: I like it
  • +18
  • Vote: I do not like it

Comments Comments (4)
| Write comment?
»
8 years ago, # |
← Rev. 2   Vote: I like it +15 Vote: I do not like it

Try getting rid of allocating O(NlogN) memory units and O(NlogN) cos/sin computations.

→ Reply
»
8 years ago, # |
  Vote: I like it +5 Vote: I do not like it

Try to search for fft on codeforces, I remember there were few topics with useful comments.

Here's one: http://mirror.codeforces.com/blog/entry/18543?#comment-235293

Also you can reduce number of calls to ifft (4 -> 1) by doing it only in the end and before just accumulating vs[i].

→ Reply
»
8 years ago, # |
  Vote: I like it 0 Vote: I do not like it

You can see my implementation of the problem using FFT.

→ Reply