速すぎる、フーリエすぎる:FFTコードゴルフ


48

できるだけ少ない文字で高速フーリエ変換を実装します。

ルール:

  • 最短のソリューションが勝つ
  • 入力は、長さが2のべき乗である1D配列であると想定できます。
  • 任意のアルゴリズムを使用できますが、解は実際には単純な離散フーリエ変換ではなく、高速フーリエ変換でなければなりません(つまり、ONログN漸近的計算コストが必要です)

編集:

  • コードは、標準の高速フーリエ変換を実装する必要があります。その形式は、このWolframの記事の式(3)で見ることができます。

    ここに画像の説明を入力してください

  • 既存の標準ライブラリまたは統計パッケージからFFT関数を使用することは許可されていません。ここでの課題は、FFTアルゴリズム自体を簡潔に実装することです。

3
これは指定不足です。少なくとも正規化係数を定義する必要があります。また、あいまいさが意図的に誤って解釈されることに注意する必要があります。たとえば、「実装」は、「FFT(3文字):標準ライブラリにあります」という答えで満たされますか?いくつかのテストケースも良いでしょう。
ピーターテイラー

出力要素の順序は重要ですか、つまり、ビット反転されたスクランブル解除を実装する必要がありますか、または出力をスクランブルされた順序のままにしておくことができますか?
ポールR

ルールの編集を参照してください。出力は、上記で参照した標準DFT式のインデックスに従って順序付けられた値を持つリスト/配列である必要があります。
jakevdp

2
実装をテストできるように、入力と出力の例を投稿できますか?
FUZxxl

2
タイトルは「Fast and Fourier-s」(Fast and Furious)である必要がありました。
clismique

回答:


12

Mathematica、95バイト

@ chyaongの助けを借りたCooley–Tukey FFTの別の実装。

{n=Length@#}~With~If[n>1,Join[+##,#-#2]&[#0@#[[;;;;2]],#0@#[[2;;;;2]]I^Array[-4#/n&,n/2,0]],#]&

非ゴルフ

FFT[x_] := With[{N = Length[x]},
  If[N > 1,
    With[{a = FFT[ x[[1 ;; N ;; 2]] ], 
          b = FFT[ x[[2 ;; N ;; 2]] ] * Table[E^(-2*I*Pi*k/N), {k, 0, N/2 - 1}]},
      Join[a + b, a - b]],
    x]]

1
私が考える#[[;;;;2]]==#[[1;;N;;2]][[2;;;;2]]==[[2;;N;;2]]
チャノッグ

1
101文字:With[{L=Length@#},If[L>1,Join[+##,#-#2]&[#0@#[[;;;;2]],#0@#[[2;;;;2]]E^(-2I*Pi(Range[L/2]-1)/L)],#]]&
チャノッグ

いいですね、再帰関数と競合することなく、その中に別の匿名関数を凝縮できます。また、パートが欠落しているインデックスを埋めることを学びました。ユニコードを使用してさらにそれを進めることができます。
マイル

9

J、37バイト

_2&(0((+,-)]%_1^i.@#%#)&$:/@|:]\)~1<#

数年後の改善。それでもCooley-Tukey FFTアルゴリズムを使用します。

使用した4バイトの保存電子πI @に= -1、おかげで漏れ修道女

オンラインでお試しください!

使用法

   f =: _2&(0((+,-)]%_1^i.@#%#)&$:/@|:]\)~1<#
   f 1 1 1 1
4 0 0 0
   f 1 2 3 4
10 _2j2 _2 _2j_2
   f 5.24626 3.90746 3.72335 5.74429 4.7983 8.34171 4.46785 0.760139
36.9894 _6.21186j0.355661 1.85336j_5.74474 7.10778j_1.13334 _0.517839 7.10778j1.13334 1.85336j5.74474 _6.21186j_0.355661

説明

_2&(0((+,-)]%_1^i.@#%#)&$:/@|:]\)~1<#  Input: array A
                                    #  Length
                                  1<   Greater than one?
_2&(                            )~     Execute this if true, else return A
_2                            ]\         Get non-overlapping sublists of size 2
    0                       |:           Move axis 0 to the end, equivalent to transpose
                          /@             Reduce [even-indexed, odd-indexed]
                       &$:               Call recursively on each 
                   #                     Get the length of the odd list
                i.@                      Range from 0 to that length exclusive
                    %#                   Divide each by the odd length
             _1^                         Compute (-1)^x for each x
           ]                             Get the odd list
            %                            Divide each in that by the previous
       +                                 Add the even values and modified odd values
         -                               Subtract the even values and modified odd values
        ,                                Join the two lists and return

1
こちらもご覧ください:blog.ndpar.com/2014/10/11/dft-j
FUZxxl

9

Python、166 151150文字

これは、基数2のCooley-Tukey FFTアルゴリズムを使用します

from math import*
def F(x):N=len(x);t=N<2or(F(x[::2]),F(x[1::2]));return N<2and x or[
a+s*b/e**(2j*pi*n/N)for s in[1,-1]for(n,a,b)in zip(range(N),*t)]

結果をテストする

>>> import numpy as np
>>> x = np.random.random(512)
>>> np.allclose(F(x), np.fft.fft(x))
True

1
2つのこと:通常はを使用するのが最適from x import*であり、のsum(([x for x in y] for y in z),[])長さよりも長くなり[x for y in z for x in y]ます。
ブースバイ

1
ありがとう-15文字節約できます!さらに11個、ツイートです。
jakevdp

ああ、それは間違いなく可能です。多くの場合、1つの改善点を見つけると、古いものが障害になります。
ブースビー

5

Pythonの3:140の 134 113文字

ショートバージョン-短くて甘い、ツイートに収まる(マイルのおかげで):

from math import*
def f(v):
 n=len(v)
 if n<2:return v
 a,b=f(v[::2])*2,f(v[1::2])*2;return[a[i]+b[i]/1j**(i*4/n)for i in range(n)]

(パイソン2では、/両側が整数であるときに分割を切り捨てている。だから、置き換える(i*4/n)ことによって(i*4.0/n)、115文字までの長さをバンプいます。)

ロングバージョン-古典的なCooley-Tukey FFTの内部をより明確に:

import cmath
def transform_radix2(vector):
    n = len(vector)
    if n <= 1:  # Base case
        return vector
    elif n % 2 != 0:
        raise ValueError("Length is not a power of 2")
    else:
        k = n // 2
        even = transform_radix2(vector[0 : : 2])
        odd  = transform_radix2(vector[1 : : 2])
        return [even[i % k] + odd[i % k] * cmath.exp(i * -2j * cmath.pi / n) for i in range(n)]

1
使用して113バイトに短縮e^(-2j * pi * i / n) = (-1)^(2 * i / n) = (1j)^(4 * i / n)
マイル

@miles非常に印象的な観察、ありがとう!10年以上にわたってDFTを繰り返し実装してきたため、sin / cos / expに夢中になり、iの単純な力を使用できることを忘れていました。私は答えを編集して新しい洞察を取り入れ、あなたに感謝します。
ナユキ

5

R:142の 133 99 95バイト

32 36バイトを削るのを手伝ってくれた@Giuseppeに感謝します!

f=function(x,n=sum(x|1),y=1:(n/2)*2)`if`(n>1,f(x[-y])+c(b<-f(x[y]),-b)*exp(-2i*(y/2-1)*pi/n),x)

ここでの追加のトリックは、メイン関数のデフォルト引数を使用していくつかの変数をインスタンス化することです。
使い方は同じです:

x = c(1,1,1,1)
f(x)
[1] 4+0i 0+0i 0+0i 0+0i

133バイトの4年前のバージョン:

f=function(x){n=length(x);if(n>1){a=Recall(x[seq(1,n,2)]);b=Recall(x[seq(2,n,2)]);t=exp(-2i*(1:(n/2)-1)*pi/n);c(a+b*t,a-b*t)}else{x}}

インデント付き:

f=function(x){
    n=length(x)
    if(n>1){
        a=Recall(x[seq(1,n,2)])
        b=Recall(x[seq(2,n,2)])
        t=exp(-2i*(1:(n/2)-1)*pi/n)
        c(a+b*t,a-b*t)
        }else{x}
    }

Cooley-Tukeyアルゴリズムも使用します。ここでの唯一のトリックは、Recall再帰性を可能にする関数の使用と、実際の計算を大幅に短縮するRベクトル化の使用です。

使用法:

x = c(1,1,1,1)
f(x)
[1] 4+0i 0+0i 0+0i 0+0i

1
4年後、それを101バイトにまで減らしました。Recall名前付き関数として既に使用している理由を100%確信することはできませんが、ちょっと後知恵でゴルフをするのは簡単です!:) +1、とてもいい。
ジュゼッペ

Recall確かに、はいは不要です。数ヶ月前に気づいたが、それを変更するには面倒だった:)私はそれを変更します。
プランナパス

非常に素晴らしい!私はさらに4バイトを絞り出した!、これをMathematicaと同等にします。
ジュゼッペ

ありがとう!私はyそこに置くことを考えましたが、それがexp(...)部品にも使用できることに気付きませんでした。
plannapus

4

Python、134

これはjakevdpのソリューションから大きく借用しているため、これをコミュニティWikiに設定しました。

from math import*
F=lambda x:x*(len(x)<2)or[a+s*b/e**(2j*pi*n/len(x))for s in(1,-1)for n,(a,b)in
enumerate(zip(F(x[::2]),F(x[1::2])))]

変更点:

-12文字:kill t

def F(x):N=len(x);t=N<2or(F(x[::2]),F(x[1::2]));return ... in zip(range(N),*t)]
def F(x):N=len(x);return ... in zip(range(N),F(x[::2]),F(x[1::2]))]

-1文字:指数のトリックx*y**-z == x/y**z (これは他の人を助けることができます)

...[a+s*b*e**(-2j*pi*n/N)...
...[a+s*b/e**(2j*pi*n/N)...

-2チャー:置き換えるand*

...return N<2and x or[
...return x*(N<2)or[

+1文字:lambdaize、killingN

def F(x):N=len(x);return x*(N<2)or[a+s*b/e**(2j*pi*n/N) ... zip(range(N) ...
F=lambda x:x*(len(x)<2)or[a+s*b/e**(2j*pi*n/len(x)) ... zip(range(len(x)) ...

-2 char:のenumerate代わりに使用zip(range(len(

...for(n,a,b)in zip(range(len(x)),F(x[::2]),F(x[1::2]))]
...for n,(a,b)in enumerate(zip(F(x[::2]),F(x[1::2])))]

しかし、これはもはや高速フーリエ変換ではないと思います...「tを殺す」ことにより、O [N log(N)]からO [N ^ 2]に移動する不要な計算を追加しました
jakevdp

私は自分の投稿に反対票を投じることはできないようです。あなたは正しいです、私はループ順序を交換し、パフォーマンスを殺しました。修正する方法を見つけた場合のために、今のところこれを残しておきます。
ブースバイ14

101バイトf=lambda x:x*(len(x)<2)or[u+v/1j**(4*i/len(x))for i,(u,v)in enumerate(zip(f(x[::2])*2,f(x[1::2])*2))]
マイル

あなたは置き換えることができfor s in(1,-1)forfor s in 1,-1for、あるいは、順序は重要ではありませんならばfor s in-1,1for
ジョナサンフレッチ

4

C、259

typedef double complex cplx;
void fft(cplx buf[],cplx out[],int n,int step){
if(step < n){
fft(out, buf,n, step * 2);
fft(out+step,buf+step,n,step*2);
for(int i=0;i<n;i+=2*step){
cplx t=cexp(-I*M_PI*i/n)*out[i+step];
buf[i/2]=out[i]+t;
buf[(i+n)/2]=out[i]-t;
}}}

問題は、そのような実装は役に立たず、簡単なアルゴリズムは非常に高速であるということです。


2
あなたは、例えば、文字の下のamoutを得るために、いくつかのより多くの空白を削除することができますstep < nに変更することができますstep<nし、step * 2に変更することができますstep*2
ProgramFOX 14

2
すべての変数と関数とのtypedefは、文字をたくさん節約するために1文字の名前を持つ必要があります

2
あなたは誰かがこれのためにたくさんの改善を提案しました。ここでそれらを見てください:codegolf.stackexchange.com/review/suggested-edits/17119
ジャスティン

1
すべての改行を削除できます。Cでは改行は使用できません
-TuxCrafting

@TùxCräftîñgすべての改行が役に立たないわけではありません。#include、#define、#ifなどのディレクティブに必要です。
Nayuki17年

3

MATLAB、128 118 107 102 101 94 93バイト

EDIT6:@algmyrに別のバイトをありがとう!

function Y=f(Y);
n=numel(Y);
k=2:2:n;
if k;
   c=f(Y(k-1));
   d=f(Y(k)).*i.^(2*(2-k)/n);
   Y=[c+d;c-d];
end

EDIT5:まだ短くなっています:)@ sanchisesのおかげで

function Y=f(Y)
n=numel(Y);
k=2:2:n;
if k;
   c=f(Y(k-1));
   d=f(Y(k)).*(-1).^((2-k)/n);
   Y=[c+d;c-d];
end

EDIT4:イェーイ、さらに-1文字(もなしで行うことができますk):

function Y=f(Y)
n=numel(Y);
if n>1;
   k=2:2:n;
   c=f(Y(k-1));
   d=f(Y(k)).*(-1).^((k/2-1)*2/n)';
   Y=[c+d;c-d];
end

EDIT2 / 3:さらに改善してくれた@sanchisesに感謝します!

function Y=f(Y)
n=numel(Y);  
if n>1;
   c=f(Y(1:2:n));
   d=f(Y(2:2:n)).*(-1).^(-(0:n/2-1)*2/n).';
   Y=[c+d;c-d]; 
end

編集:いくつかの改善を行うことができ、スケーリング定数が必要ないことに気づいた。

これは拡張バージョンです。改行/スペースを削除すると、文字カウントが有効になります。(列ベクトルに対してのみ機能します。)

function y=f(Y)
n=numel(Y);  
y=Y;
if n>1;
   c=f(Y(1:2:n));
   d=f(Y(2:2:n));
   n=n/2;
   d=d.*exp(-pi*i*(0:n-1)/n).';
   y=[c+d;c-d]; 
end

ヒント:2つのd=行を1 つに結合できますm=n/2;d=f(Y(2:2:n)).*exp(-pi*i*(0:m-1)/m).';。さらに、3行目に変更y=f(Y)Y=f(Y)て削除することを検討してください(コードゴルフ以外では絶対にしないことをお約束します)
-Sanchises

ああ、ありがとう!function Y = f(Y)読み難さ以外の欠点はありますか?
flawr

まあ、MATLABは、Yが変更されない場合でも、戻り値について文句を言うことはありません。それは少し速いので、いくつかの目的(つまり、入力変数をほとんど変更しない関数)のために結局それほど悪くないと思います
-Sanchises

さて、さらにシェービングをオフにするために:m=n/2を削除し、代わりmn/2とでn*2それぞれ置き換えることができます。そして、私はプログラムがMATLABで可能な限り短いと強く信じています。
-Sanchises

1
そして、私はプログラムがMATLABで可能な限り短いと強く信じています。-Sanchises 15年3月8日21:05で有名な最後の言葉
...-Sanchises

2

ゼリー、31 30 28 26 バイト非競合

LḶ÷$N-*×,N$+ḷF
s2Z߀ç/µ¹Ṗ?

ゼリーはこのチャレンジの後に作成されたため、競合しません。

これは、Cooley-Tukey radix-2再帰アルゴリズムを使用します。ゴルフされていないバージョンについては、Mathematicaでの私の答えをご覧ください。

オンラインで試すか、複数のテストケースを確認します

説明

LḶ÷$N-*×,N$+ḷF  Helper link. Input: lists A and B
L               Get the length of A
   $            Operate on that length
 Ḷ                Make a range [0, 1, ..., length-1]
  ÷               Divide each by length
    N           Negate each
     -          The constant -1
      *         Compute -1^(x) for each x in that range
       ×        Multiply elementwise between that range and B, call it B'  
          $     Operate on that B'
         N        Negate each
        ,         Make a list [B', -B']
            ḷ   Get A
           +    Add vectorized, [B', -B'] + A = [A+B', A-B']
             F  Flatten that and return

s2Z߀ç/µ¹Ṗ?  Main link. Input: list X
         Ṗ   Curtail - Make a copy of X with the last value removed
          ?  If that list is truthy (empty lists are falsey)
       µ       Parse to the left as a monad
s2             Split X into sublists of length 2
  Z            Transpose them to get [even-index, odd-index]
   ߀          Call the main link recursively on each sublist
     ç/        Call the helper link as a dyad on the sublists and return
             Else
        ¹      Identity function on X and return

2

C(gcc)188186184184183バイト

#define d(a,b,c)f(a,b,c,1,0)
f(a,b,c,n,k)_Complex*a,*b;{_Complex z[c];*b=*a;if(n<c)for(f(a,z,c,n*2),f(a+n,z+n,c,n*2);k<c;k+=n*2)b[k+c>>1]=z[k]*2-(b[k/2]=z[k]+z[k+n]/cpow(1i,2.*k/c));}

オンラインでお試しください!

わずかにゴルフが少ない

#define d(a,b,c)f(a,b,c,1,0)
f(a,b,c,n,k)_Complex*a,*b;{
  _Complex z[c];
  *b=*a;
  if(n<c)
    for(f(a,z,c,n*2),f(a+n,z+n,c,n*2);k<c;k+=n*2)
      b[k+c>>1]=z[k]*2-(b[k/2]=z[k]+z[k+n]/cpow(1i,2.*k/c));
}

1

パリ/ GP、76文字

X(v)=my(t=-2*Pi*I/#v,s);vector(#v,k,s=t*(k-1);sum(n=0,#v-1,v[n+1]*exp(s*n)))

使用法

X([1,1,1,1])
%2 = [4.000000000000000000000000000, 0.E-27 + 0.E-28*I, 0.E-28 + 0.E-27*I, 0.E-27 + 0.E-28*I]

3
これは単純なDFTではありませんか?(すなわちtheta(N ^ 2))
マイル

1

オクターブ109103101100バイト

f(f=@(f)@(x,n=rows(x)){@(d=f(f)(x(k=2:2:n)).*i.^((k*2-4)/n)')[d+(c=f(f)(x(k-1)));c-d],x}{1+(n<2)}())

オンラインでお試しください!

Oooooは、この再帰的な呪われたラムダから目を出血させます。この大部分は@flawrの回答から取り除かれました。

f(                                          % lambda function
  f=@(f)                                    % defined in its own argument list, 
                                            % accepts itself as parameter (for recursion)
    @(x,n=rows(x)){                         % calls another lambda,
                                            % 2nd parameter is just to define a variable
      @(d=f(f)(x(k=2:2:n)).*i.^((k*2-4)/n)')% 1/4 of FFT (argument just defines a variable)
        [d+(c=f(f)(x(k-1)));                % 2/4 of FFT
         c-d                                % 4/4 of FFT
        ],                                  % This is in a @()[] to inhibit evaluation
                                            % unless actually called
      x                                     % FFT of length 1
    }{1+(n<2)}                              % if len(x)==1, return x
                                            % else return [d+c;c-d]
  ()                                        % this is probably important too
)

私はあなたがしたことを理解していませんが、私はそれがとても好きです。
flawr

0

公理、259193181、179のバイト

L(g,n,f)==>[g for i in 1..n|f]
h(a)==(n:=#a;n=1=>a;c:=h(L(a.i,n,odd? i));d:=h(L(a.i,n,even? i));n:=n/2;t:=1>0;v:=L(d.i*%i^(-2*(i-1)/n),n,t);append(L(c.i+v.i,n,t),L(c.i-v.i,n,t)))

h(a)がすべてのテストに合格し、この「競合」のエントリとしては問題ない場合でも、argumentsチェックするために、以下のfft()を介してh()またはhlp()を呼び出す必要があります。私は他の人が書いたものを見ただけなので、このソフトウェアが大丈夫かどうかわかりません。以下のコメントなしのコードなしのコード:

-- L(g,n,f)==>[g for i in 1..n|f]
-- this macro L, build one List from other list, where in g, there is the generic element of index i
-- (as a.i, or a.i*b.i or a.i*4), n build 1..n that is the range of i, f is the condition 
-- for insert the element in the list result.

hlp(a)==
    n:=#a;n=1=>a
    -- L(a.i,n,odd? i)  it means build a list getting "even indices i of a.i as starting from index 0" [so even is odd and odd is even]
    -- L(a.i,n,even? i) it means build a list getting "odd  indices i of a.i as starting from index 0"
    c:=hlp(L(a.i,n,odd? i));d:=hlp(L(a.i,n,even? i))
    n:=n/2;t:=1>0
    v:=L(d.i*%i^(-2*(i-1)/n),n,t)
    append(L(c.i+v.i,n,t),L(c.i-v.i,n,t))

-- Return Fast Fourier transform of list a, in the case #a=2^n
fft(a)==(n:=#a;n=0 or gcd(n,2^30)~=n=>[];hlp(a))

(5) -> h([1,1,1,1])
   (5)  [4,0,0,0]
                                    Type: List Expression Complex Integer
(6) -> h([1,2,3,4])
   (6)  [10,- 2 + 2%i,- 2,- 2 - 2%i]
                                    Type: List Expression Complex Integer
(7) -> h([5.24626,3.90746,3.72335,5.74429,4.7983,8.34171,4.46785,0.760139])
   (7)
   [36.989359, - 6.2118552150 341603904 + 0.3556612739 187363298 %i,
    1.85336 - 5.744741 %i, 7.1077752150 341603904 - 1.1333387260 812636702 %i,
    - 0.517839, 7.1077752150 341603904 + 1.1333387260 812636702 %i,
    1.85336 + 5.744741 %i,
    - 6.2118552150 341603904 - 0.3556612739 187363298 %i]
                                      Type: List Expression Complex Float
(8) -> h([%i+1,2,%i-2,9])
   (8)  [10 + 2%i,3 + 7%i,- 12 + 2%i,3 - 7%i]
                                    Type: List Expression Complex Integer

いくつかでは、h()またはfft()が正確な解を返すのを見ましたが、単純化がうまくいかない場合:

(13) -> h([1,2,3,4,5,6,7,8])
   (13)
                    +--+                                   +--+
        (- 4 + 4%i)\|%i  - 4 + 4%i             (- 4 - 4%i)\|%i  - 4 + 4%i
   [36, --------------------------, - 4 + 4%i, --------------------------, - 4,
                    +--+                                   +--+
                   \|%i                                   \|%i
            +--+                                   +--+
    (- 4 + 4%i)\|%i  + 4 - 4%i             (- 4 - 4%i)\|%i  + 4 - 4%i
    --------------------------, - 4 - 4%i, --------------------------]
                +--+                                   +--+
               \|%i                                   \|%i
                                    Type: List Expression Complex Integer

以下の8のように、リストの1つの要素の型のみを変更するだけで十分です。

(14) -> h([1,2,3,4,5,6,7,8.])
   (14)
   [36.0, - 4.0000000000 000000001 + 9.6568542494 923801953 %i, - 4.0 + 4.0 %i,
    - 4.0 + 1.6568542494 92380195 %i, - 4.0, - 4.0 - 1.6568542494 92380195 %i,
    - 4.0 - 4.0 %i, - 4.0 - 9.6568542494 923801953 %i]
                                      Type: List Expression Complex Float

私はそれを書いたが、他のすべての回答を見たのは、リンクではページが非常に難しかったので、このコードが正しいかどうかわからないからだ。私は一人のFFTエキスパートではないので、これはすべて(おそらく)間違っている可能性があります。


0

APL(NARS)、58文字、116バイト

{1≥k←≢⍵:⍵⋄(∇⍵[y∼⍨⍳k])(+,-)(∇⍵[y←2×⍳t])×0J1*t÷⍨2-2×⍳t←⌊k÷2}

テスト

  f←{1≥k←≢⍵:⍵⋄(∇⍵[y∼⍨⍳k])(+,-)(∇⍵[y←2×⍳t])×0J1*t÷⍨2-2×⍳t←⌊k÷2}
  f 1 1 1 1
4J0 0J0 0J0 0J0 
  f 1 2 3 4
10J0 ¯2J2 ¯2J0 ¯2J¯2 
  f 1J1 2 ¯2J1  9
10J2 3J7 ¯12J2 3J¯7 
  f 5.24626,3.90746,3.72335,5.74429,4.7983,8.34171,4.46785,0.760139
36.989359J0 ¯6.211855215J0.3556612739 1.85336J¯5.744741 7.107775215J¯1.133338726 ¯0.517839J0 
  7.107775215J1.133338726 1.85336J5.744741 ¯6.211855215J¯0.3556612739 
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.