結果

問題 No.2807 Have Another Go (Easy)
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-09-11 00:27:13
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,705 bytes
コンパイル時間 200 ms
コンパイル使用メモリ 82,428 KB
実行使用メモリ 86,152 KB
最終ジャッジ日時 2024-09-11 00:27:23
合計ジャッジ時間 9,065 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,888 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 AC 101 ms
76,628 KB
testcase_38 WA -
testcase_39 WA -
testcase_40 AC 156 ms
77,712 KB
testcase_41 AC 162 ms
78,168 KB
testcase_42 AC 156 ms
78,052 KB
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1331

MOD = 998244353

def main():
    N, M, k = map(int, input().split())
    C = list(map(int, input().split()))

    # 何も障害がない場合の回答を用意
    max_x = N * M + 6
    dp = [0] * max_x
    dp[0] = 1
    for i in range(N * M):
        for j in range(1, 7):
            dp[i + j] += dp[i]
            dp[i + j] %= MOD

    total_value = 0
    for j in range(6):
        total_value += dp[N * M + j]
        total_value %= MOD

    # 各C_iについて回答を用意
    for c in C:
        dp1 = {}
        for j in range(1, 6):
            if c - j < 0:
                continue
            dp1[c - j] = dp[c - j]
        
        dp2 = {}
        for j in range(1, 6):
            if (c - j) in dp1:
                for k in range(1, 7):
                    if (c - j + k) > c:
                        if (c - j + k) not in dp2:
                            dp2[(c - j + k)] = 0
                        dp2[(c - j + k)] += dp1[c - j]
                        dp2[(c - j + k)] %= MOD

        dp3 = {}
        for key, value in dp2.items():
            for j in range(1, 6):
                if N + c - j < key:
                    continue
                if (N + c - j) not in dp3:
                    dp3[(N + c - j)] = 0
                dp3[(N + c - j)] += (value * dp[(N + c - j - key)])% MOD
                dp3[(N + c - j)] %= MOD
        
        dp4 = {}
        for j in range(1, 6):
            if (N + c - j) in dp3:
                for k in range(1, 7):
                    if (N + c - j + k) >= N * M:
                        continue
                    elif (N + c - j + k) > N + c:
                        if (N + c - j + k) not in dp4:
                            dp4[(N + c - j + k)] = 0
                        dp4[(N + c - j + k)] += dp3[N + c - j]
                        dp4[(N + c - j + k)] %= MOD

        dp5 = {}
        for key, value in dp4.items():
            for j in range(1, 6):
                if 2 * N - j < key:
                    continue

                if (2 * N - j) not in dp5:
                    dp5[(2 * N - j)] = 0
                dp5[(2 * N - j)] += (value * dp[(2 * N - j - key)])% MOD
                dp5[2 * N - j] %= MOD

        answer = 0
        for key, value in dp5.items():
            for j in range(1, 7):
                if key + j >= N * M:
                    answer += value
                    answer %= MOD
        for key, value in dp3.items():
            for j in range(1, 7):
                if key + j >= N * M:
                    answer += value
                    answer %= MOD
        print((total_value - answer) % MOD)











if __name__ == "__main__":
    main()
0