python-091Đọc toàn bộ đề miễn phí

Biến đổi Fourier Nhanh 1D (Cooley-Tukey FFT) đệ quy thuần NumPy

Viết hàm fft_1d(x: np.ndarray) -> np.ndarray:

PythonNâng cao35 phút

Tiến độ của tôi ở bài này

Điểm được lưu vào tài khoản sau khi chấm bài.

Đang tải điểm của bạn…

Kiến thức và chủ đề

fftsignal processingrecursionnumpy

Kiến thức tiên quyết: complex numbers, recursion.

Nội dung đề bài

Mục tiêu kiến thức

  • Hiểu giải thuật Chia để trị (Divide and Conquer) trong Biến đổi Fourier Nhanh (FFT) độ phức tạp O(N log N).
  • Phân tách chuỗi chẵn và chuỗi lẻ (Decimation in Time).
  • Vector hóa thao tác cánh bướm (Butterfly Operation) với hệ số xoay (Twiddle Factors) số phức WNk = e-2π i k / N.

Mô tả bài toán

Viết hàm fft_1d(x: np.ndarray) -> np.ndarray:

  • x: Mảng 1D các số thực hoặc phức.
  • Kiểm tra độ dài N = len(x):
  • Nếu N không phải là lũy thừa của 2 (N ≤ 0 hoặc (N & (N - 1)) ≠ 0), raise ValueError("Do dai mang phai la luy thua cua 2").
  • Điều kiện dừng đệ quy:
  • Nếu N = 1, trả về chính mảng x.astype(np.complex128).
  • Đệ quy:
  • Phân tách: gọi đệ quy trên các phần tử chỉ số chẵn Xeven = fft_1d(x[0::2]) và chỉ số lẻ Xodd = fft_1d(x[1::2]).
  • Tính hệ số xoay cho nửa đầu k = 0, 1, …, N/2 - 1:

Tk = e-2π i k / N · Xodd[k].

  • Tổng hợp kết quả cánh bướm:
  • Nửa đầu: X[k] = Xeven[k] + Tk
  • Nửa sau: X[k + N/2] = Xeven[k] - Tk
  • Trả về mảng kết quả 1D kiểu np.complex128.

Input

  • Tham số: x: np.ndarray.

Output

  • Trả về: np.ndarray chứa các số phức complex128.

Ràng buộc

  • Thời gian chạy tối đa: 1000ms.
  • Giới hạn bộ nhớ: 256MB.
  • Dữ liệu đầu vào tuân thủ đúng kiểu dữ liệu và miền giá trị được mô tả.

Ví dụ 1

Input

fft_1d(x=[1.0, 1.0, 1.0, 1.0])

Output

[[4.0, 0.0], [0.0, 0.0], [0.0, 0.0], [0.0, 0.0]]

Giải thích

Hàm được gọi với các tham số mẫu trên và trả về kết quả chính xác theo yêu cầu.

Ví dụ 2

Input

fft_1d(x=[1.0, 0.0, 1.0, 0.0])

Output

[[2.0, 0.0], [0.0, 0.0], [2.0, 0.0], [0.0, 0.0]]

Giải thích

Hàm được gọi với bộ tham số thứ hai và trả về kết quả tương ứng theo thiết kế.

3 cấp độ gợi ýMở dần khi bạn thật sự cần hỗ trợ.
Phân tích lời giảiGiải thích hướng tư duy và thuật toán.
Code tham khảoDùng để đối chiếu sau khi tự làm.

Gợi ý và lời giải chỉ mở sau khi bạn bấm Nộp bài. Giáo viên và quản trị viên mở được ngay.