結果

問題 No.2807 Have Another Go (Easy)
コンテスト
ユーザー detteiuu
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0