結果
問題 | 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 = 998244353dp0 = [0]*(N+1)dp0[0] = 1for i in range(N):for j in range(1, 7):if i+j <= N:dp0[i+j] += dp0[i]dp0[i+j] %= MODcntN = dp0[-1]dp = [0]*(N*M+1)dp[0] = 1dp2 = [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)] %= MODdp2[min(i+j, N*M)] += dp[i]*(7-j)dp2[min(i+j, N*M)] %= MODdp2 = dp2[::-1]for i in range(K):top = dp[C[i]]*dp2[C[i]]%MODbottom = dp[C[i]+N]*dp2[C[i]+N]%MODminus = cntN*dp[C[i]]*dp2[C[i]+N]%MODprint((top+bottom-minus)%MOD)