結果
問題 |
No.1832 NAND Reversible
|
ユーザー |
|
提出日時 | 2022-02-04 23:35:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 74 ms / 2,000 ms |
コード長 | 611 bytes |
コンパイル時間 | 175 ms |
コンパイル使用メモリ | 82,192 KB |
実行使用メモリ | 79,360 KB |
最終ジャッジ日時 | 2024-06-11 13:01:40 |
合計ジャッジ時間 | 2,212 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
N,K = map(int,input().split()) import sys P = 998244353 C = N + 5 fact = [1] * C fact_inv = [1] * C for i in range(2,C): fact[i] = fact[i-1] * i % P fact_inv[-1] = pow(fact[-1],P-2,P) for i in range(C-2,0,-1): fact_inv[i] = fact_inv[i+1] * (i + 1) % P def comb(a,b): return fact[a] * fact_inv[b] * fact_inv[a-b] % P if K == 0: print(1) exit() if K == 1: if (N-1) % 2 == 1: print(2) else: print(N-2) exit() ans = comb(N-2,N-K) for i in range(1,(N-K)//2+1): ans = (ans + 2 * comb(N-2*i-2,K-2)) % P ans = (ans + comb(N-2*i-2,K-2) * (2*i-1))%P print(ans)