結果
問題 |
No.723 2つの数の和
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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])