結果
問題 | No.786 京都大学の過去問 |
ユーザー | vwxyz |
提出日時 | 2021-10-08 03:26:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 80 ms / 2,000 ms |
コード長 | 960 bytes |
コンパイル時間 | 355 ms |
コンパイル使用メモリ | 87,216 KB |
実行使用メモリ | 71,528 KB |
最終ジャッジ日時 | 2023-09-30 09:13:11 |
合計ジャッジ時間 | 1,291 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 75 ms
71,140 KB |
testcase_01 | AC | 74 ms
71,368 KB |
testcase_02 | AC | 77 ms
71,528 KB |
testcase_03 | AC | 80 ms
71,472 KB |
testcase_04 | AC | 76 ms
71,316 KB |
ソースコード
def Bostan_Mori(poly_nume,poly_deno,N,mod=0,fft=False,ntt=False): def convolve(poly_nume,poly_deno): conv=[0]*(len(poly_nume)+len(poly_deno)-1) for i in range(len(poly_nume)): for j in range(len(poly_deno)): conv[i+j]+=poly_nume[i]*poly_deno[j] if mod: for i in range(len(conv)): conv[i]%=mod return conv while N: poly_deno_=[-x if i%2 else x for i,x in enumerate(poly_deno)] if N%2: poly_nume=convolve(poly_nume,poly_deno_)[1::2] else: poly_nume=convolve(poly_nume,poly_deno_)[::2] poly_deno=convolve(poly_deno,poly_deno_)[::2] if fft and mod: for i in range(len(poly_nume)): poly_nume[i]%=mod for i in range(len(poly_deno)): poly_deno[i]%=mod N//=2 return poly_nume[0] N=int(input()) ans=Bostan_Mori([1,0],[1,-1,-1],N) print(ans)