B4: Quantum Fourier Transform

実行時間制限:3 秒

メモリ制限:512 MiB

配点:200点

Writer:PSL

解説

問題のとおり、量子フーリエ変換は次式の遷移を満たすオラクルとして定義されます。

∣j⟩n→QFT12n∑k=02n−1exp⁡(2πijk2n)∣k⟩n\begin{equation} \ket{j}_n \xrightarrow{\mathrm{QFT}} \sqrt{\frac{1}{2^n}} \sum_{k=0}^{2^n-1} \exp{\left(\frac{2\pi i j k}{2^n}\right)} \ket{k}_n \end{equation}

このとき、右辺は、正規化係数 1/2n1/\sqrt{2^n} を除くと、次のように各量子ビットごとに独立な状態の積として表すことができます。

∑k=02n−1exp⁡(2πijk2n)∣k⟩n=∑k0=01⋯∑kn−1=01exp⁡(2πij∑l=0n−1(2lkl)2n)∣k0⋯kn−1⟩=∑k0=01⋯∑kn−1=01exp⁡(2πij∑l=0n−1(2l−nkl))∣k0⋯kn−1⟩=(∑k0=01e2πij2−nk0∣k0⟩)⊗(∑k1=01e2πij21−nk1∣k1⟩)⊗⋯⊗(∑kn−1=01e2πij2−1kn−1∣kn−1⟩)=(∣0⟩+e2πi0.jn−1⋯j0∣1⟩)⊗(∣0⟩+e2πi0.jn−2⋯j0∣1⟩)⊗⋯⊗(∣0⟩+e2πi0.j0∣1⟩)\begin{align} \sum_{k=0}^{2^n-1} \exp{\left(\frac{2\pi i j k}{2^n}\right)} \ket{k}_n &= \sum_{k_0=0}^{1} \cdots \sum_{k_{n-1}=0}^{1} \exp{\left(\frac{2\pi i j \sum_{l=0}^{n-1} (2^l k_l)}{2^n}\right)} \ket{k_0 \cdots k_{n-1}} \nonumber\\ &= \sum_{k_0=0}^{1} \cdots \sum_{k_{n-1}=0}^{1} \exp{\left(2\pi i j \sum_{l=0}^{n-1} (2^{l-n} k_l)\right)} \ket{k_0 \cdots k_{n-1}} \nonumber\\ &= \left(\sum_{k_0=0}^{1} e^{2\pi i j 2^{-n} k_0}\ket{k_0}\right) \otimes \left(\sum_{k_1=0}^{1} e^{2\pi i j 2^{1-n} k_1}\ket{k_1}\right)\otimes \cdots \otimes \left(\sum_{k_{n-1}=0}^{1} e^{2\pi i j 2^{-1} k_{n-1}}\ket{k_{n-1}}\right) \nonumber\\ &= (\ket{0} + e^{2\pi i 0.j_{n-1} \cdots j_0}\ket{1}) \otimes (\ket{0} + e^{2\pi i 0.j_{n-2} \cdots j_0}\ket{1}) \otimes \cdots \otimes (\ket{0} + e^{2\pi i 0.j_{0}}\ket{1}) \end{align}

ただし、0.x0x1⋯xm=∑k=0m2−(k+1)xk0.x_0x_1 \cdots x_m = \sum_{k=0}^{m} 2^{-(k+1)} x_k は二進法の小数表記を表します。

(2) 式の操作は、次の量子回路(n=3n=3)のようにアダマールゲートと制御位相ゲート、スワップゲートを用いて実装できます。

QCoder・Qiskit ではリトルエンディアンを採用しているため、一般的な量子回路図 とはビットの並びが逆になっていることに注意して下さい。

例えば、この図で 33 番目の量子ビットの状態は次のように遷移します。

∣j2⟩→H(∣0⟩+e2πi0.j2∣1⟩)/2→CP(π/2,1,2)(∣0⟩+e2πi0.j2j1∣1⟩)/2→CP(π/4,0,2)(∣0⟩+e2πi0.j2j1j0∣1⟩)/2\begin{align} \ket{j_2} &\xrightarrow{H} (\ket{0} + e^{2\pi i 0.j_2}\ket{1})/\sqrt{2}\nonumber\\ &\xrightarrow{CP(\pi/2, 1, 2)} (\ket{0} + e^{2\pi i 0.j_2j_1}\ket{1})/\sqrt{2}\nonumber\\ &\xrightarrow{CP(\pi/4, 0, 2)} (\ket{0} + e^{2\pi i 0.j_2j_1j_0}\ket{1})/\sqrt{2} \end{align}

最後にスワップゲートによってこの状態は 11 番目の量子ビットと交換されます。 こうして最終的な 11 番目の量子ビットの状態は (∣0⟩+e2πi0.j2j1j0∣1⟩)/2(\ket{0} + e^{2\pi i 0.j_2j_1j_0}\ket{1})/\sqrt{2} となります。 その他の量子ビットについても最終的な状態が (2) 式と一致するかぜひ確かめてみてください。

ゲートを左に詰めていくことでこの量子回路の深さは 2n2n とわかります。

量子フーリエ変換はショアのアルゴリズムや量子位相推定などの量子アルゴリズムに用いられる重要なサブルーチンとして知られています。

writer 解説

解答例

解答例は以下の通りです。

import math
 
from qiskit import QuantumCircuit
 
 
def solve(n: int) -> QuantumCircuit:
    qc = QuantumCircuit(n)
 
    for i in reversed(range(n)):
        qc.h(i)
        for j in reversed(range(i)):
            qc.cp(math.pi / 2 ** (i - j), j, i)
 
    for i in range(n // 2):
        qc.swap(i, n - i - 1)
 
    return qc