結果
問題 |
No.1967 Sugoroku Optimization
|
ユーザー |
![]() |
提出日時 | 2022-11-20 16:04:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 221 ms / 2,000 ms |
コード長 | 588 bytes |
コンパイル時間 | 171 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 107,648 KB |
最終ジャッジ日時 | 2024-09-21 15:32:26 |
合計ジャッジ時間 | 4,105 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
N, K = map(int, input().split()) mod = 998244353 inv = [pow(i,mod-2,mod) for i in range(N+1)] dp = [[0]*(N+2) for _ in range(K+1)] dp[0][0] = 1 dp[0][1] = -1 for i in range(1,K+1): for j in range(1,N+2): dp[i-1][j] += dp[i-1][j-1] dp[i-1][j] %= mod for j in range(N+1): if dp[i-1][j]>0: dp[i][j+1] += inv[N-j]*dp[i-1][j] dp[i][j+1] %= mod dp[i][-1] -= inv[N-j]*dp[i-1][j] dp[i][-1] %= mod res = 0 for i in range(1,N): dp[-1][i] += dp[-1][i-1] dp[-1][i] %= mod res += dp[-1][i] print((1-res)%mod)