結果
問題 | No.2791 Beginner Contest |
ユーザー | ゼット |
提出日時 | 2024-06-21 21:31:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 114 ms / 2,000 ms |
コード長 | 365 bytes |
コンパイル時間 | 205 ms |
コンパイル使用メモリ | 82,392 KB |
実行使用メモリ | 77,556 KB |
最終ジャッジ日時 | 2024-06-21 21:31:12 |
合計ジャッジ時間 | 2,489 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
N,K=map(int,input().split()) u=[1]*(N+1) mod=998244353 for i in range(1,N+1): u[i]=u[i-1]*i u[i]%=mod u2=[1]*(N+1) for i in range(N+1): u2[i]=pow(u[i],-1,mod) def ncm(x,y): ans=u[x]*u2[y] ans%=mod ans*=u2[x-y] ans%=mod return ans result=1 for k in range(1,N+1): rest=N-k*K if rest<0: break result+=ncm(rest+k,k) result%=mod print(result)