結果
問題 |
No.287 場合の数
|
ユーザー |
|
提出日時 | 2020-09-25 18:29:09 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 529 ms / 5,000 ms |
コード長 | 335 bytes |
コンパイル時間 | 294 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 44,320 KB |
最終ジャッジ日時 | 2024-06-28 05:55:27 |
合計ジャッジ時間 | 15,780 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
import numpy as np from numpy.fft import rfft,irfft def convolve(a1,a2): n = 2**(len(a1)+len(a2)+1).bit_length() b1,b2=np.zeros(n),np.zeros(n) b1[:len(a1)], b2[:len(a2)]= a1,a2 return (irfft(rfft(b1)*rfft(b2))+0.5).astype(int) n=int(input()) f = np.ones(n+1) for i in range(3): f = convolve(f,f) print(f[6*n])