結果

問題 No.723 2つの数の和
ユーザー Navier_Boltzmann
提出日時 2021-09-25 19:27:29
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 693 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 70,192 KB
最終ジャッジ日時 2024-07-05 14:30:08
合計ジャッジ時間 17,040 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

import numpy as np

def convolve(f, g):
    tf = np.array(f, np.int64)
    tg = np.array(g, np.int64) 
    fft_len = 1
    while 2 * fft_len < len(tf) + len(tg) - 1:
        fft_len *= 2
    
    fft_len *= 2

    # フーリエ変換
    Ff = np.fft.rfft(tf, fft_len)
    Fg = np.fft.rfft(tg, fft_len)

    # 各点積
    Fh = Ff * Fg

    # フーリエ逆変換
    h = np.fft.irfft(Fh, fft_len)

    # 小数になっているので、整数にまるめる
    h = np.rint(h).astype(np.int64)

    return list(h[:len(f) + len(g) - 1])
    
N,X = map(int,input().split())
a = list(map(int,input().split()))
f = [0]*(max(a)+1)
for i in range(N):
  f[a[i]] += 1
H = convolve(f,f)
print(H[X])
0