結果

問題 No.2807 Have Another Go (Easy)
ユーザー detteiuu
提出日時 2024-07-12 22:11:29
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 671 bytes
コンパイル時間 491 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 104,296 KB
最終ジャッジ日時 2024-07-12 22:11:42
合計ジャッジ時間 7,376 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 12 WA * 34
権限があれば一括ダウンロードができます

ソースコード

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