結果
問題 | No.2951 Similar to Mex |
ユーザー |
![]() |
提出日時 | 2024-09-13 01:11:00 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 377 ms / 2,000 ms |
コード長 | 947 bytes |
コンパイル時間 | 176 ms |
コンパイル使用メモリ | 82,380 KB |
実行使用メモリ | 112,432 KB |
最終ジャッジ日時 | 2024-10-25 20:50:43 |
合計ジャッジ時間 | 8,813 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
mod = 998244353 N, M, K = map(int, input().split()) S = [[0] * (N + 1) for i in range(N + 1)] for i in range(1, N + 1): S[i][1] = 1 for j in range(2, i + 1): S[i][j] = (S[i - 1][j - 1] + S[i - 1][j] * j) % mod fac = [1] * (N + 1) for i in range(1, N + 1): fac[i] = fac[i - 1] * i % mod dp = [[0] * (N + 1) for i in range(M + 2)] val = 1 for i in range(M + 1, K + 1): val *= i val %= mod dp[M + 1][0] = val for i in range(M, 0, -1): ndp = [[0] + dp[j][:N] for j in range(M + 2)] for j in range(i + 1, M + 2): for k in range(N + 1): ndp[i][k] += dp[j][k] ndp[i][k] %= mod if i <= K: for j in range(i, M + 2): for k in range(N + 1): ndp[j][k] *= j ndp[j][k] %= mod dp = ndp ans = 0 for s in range(1, N + 1): for i in range(1, M + 2): ans += dp[i][s] * S[N][s] % mod * fac[s] ans %= mod print(ans)