結果
問題 | No.786 京都大学の過去問 |
ユーザー | vwxyz |
提出日時 | 2021-10-08 03:26:18 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 33 ms / 2,000 ms |
コード長 | 960 bytes |
コンパイル時間 | 239 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 10,880 KB |
最終ジャッジ日時 | 2024-07-23 03:16:08 |
合計ジャッジ時間 | 1,007 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 33 ms
10,624 KB |
testcase_01 | AC | 32 ms
10,752 KB |
testcase_02 | AC | 31 ms
10,752 KB |
testcase_03 | AC | 30 ms
10,752 KB |
testcase_04 | AC | 30 ms
10,880 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)