結果
問題 | No.2807 Have Another Go (Easy) |
ユーザー |
![]() |
提出日時 | 2024-07-12 22:12:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 213 ms / 3,000 ms |
コード長 | 677 bytes |
コンパイル時間 | 197 ms |
コンパイル使用メモリ | 82,392 KB |
実行使用メモリ | 104,064 KB |
最終ジャッジ日時 | 2024-07-12 22:12:10 |
合計ジャッジ時間 | 7,216 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 46 |
ソースコード
N, M, K = map(int, input().split()) C = list(map(int, input().split())) MOD = 998244353 dp0 = [0]*(N+1) dp0[0] = 1 for i in range(N): for j in range(1, 7): if i+j <= N: dp0[i+j] += dp0[i] dp0[i+j] %= MOD cntN = dp0[-1] dp = [0]*(N*M+1) dp[0] = 1 dp2 = [0]*(N*M+1) for i in range(N*M): for j in range(1, 7): dp[min(i+j, N*M)] += dp[i] dp[min(i+j, N*M)] %= MOD dp2[min(i+j, N*M)] += dp[i]*(7-j) dp2[min(i+j, N*M)] %= MOD dp2 = dp2[::-1] for i in range(K): top = dp[C[i]]*dp2[C[i]]%MOD bottom = dp[C[i]+N]*dp2[C[i]+N]%MOD minus = cntN*dp[C[i]]*dp2[C[i]+N]%MOD print((top+bottom-minus)%MOD)