結果

問題 No.2807 Have Another Go (Easy)
ユーザー navel_tosnavel_tos
提出日時 2024-07-12 22:29:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 118 ms / 3,000 ms
コード長 2,416 bytes
コンパイル時間 229 ms
コンパイル使用メモリ 82,564 KB
実行使用メモリ 84,628 KB
最終ジャッジ日時 2024-07-12 22:30:06
合計ジャッジ時間 5,859 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
54,304 KB
testcase_01 AC 113 ms
83,788 KB
testcase_02 AC 97 ms
81,316 KB
testcase_03 AC 94 ms
80,556 KB
testcase_04 AC 69 ms
79,180 KB
testcase_05 AC 110 ms
82,252 KB
testcase_06 AC 104 ms
82,156 KB
testcase_07 AC 71 ms
75,708 KB
testcase_08 AC 82 ms
78,344 KB
testcase_09 AC 82 ms
78,732 KB
testcase_10 AC 80 ms
79,628 KB
testcase_11 AC 115 ms
84,280 KB
testcase_12 AC 114 ms
84,552 KB
testcase_13 AC 112 ms
84,180 KB
testcase_14 AC 111 ms
84,628 KB
testcase_15 AC 116 ms
84,300 KB
testcase_16 AC 117 ms
84,068 KB
testcase_17 AC 115 ms
84,552 KB
testcase_18 AC 115 ms
84,508 KB
testcase_19 AC 116 ms
84,296 KB
testcase_20 AC 115 ms
84,096 KB
testcase_21 AC 114 ms
84,100 KB
testcase_22 AC 113 ms
84,312 KB
testcase_23 AC 112 ms
84,212 KB
testcase_24 AC 112 ms
84,492 KB
testcase_25 AC 114 ms
84,084 KB
testcase_26 AC 115 ms
84,276 KB
testcase_27 AC 117 ms
84,436 KB
testcase_28 AC 118 ms
84,292 KB
testcase_29 AC 115 ms
84,284 KB
testcase_30 AC 115 ms
84,276 KB
testcase_31 AC 45 ms
60,396 KB
testcase_32 AC 44 ms
60,480 KB
testcase_33 AC 46 ms
61,336 KB
testcase_34 AC 44 ms
61,940 KB
testcase_35 AC 45 ms
61,024 KB
testcase_36 AC 60 ms
66,916 KB
testcase_37 AC 47 ms
60,888 KB
testcase_38 AC 72 ms
76,376 KB
testcase_39 AC 42 ms
59,152 KB
testcase_40 AC 64 ms
73,992 KB
testcase_41 AC 73 ms
76,352 KB
testcase_42 AC 76 ms
76,408 KB
testcase_43 AC 70 ms
73,748 KB
testcase_44 AC 68 ms
74,664 KB
testcase_45 AC 74 ms
72,480 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#yukicoder 2807 Have Another Go(Easy)

#入力受取
N, M, K = map(int, input().split())
C = list(map(int, input().split()))
MOD = 998244353

#独立に問題を解く場合
def brute(N, M, Ci):
    #DP[x][f]: 変数がx, Ci mod Nが出ていればf == 1となる場合の数
    DP = [[0] * 2 for x in range(N * M + 1)]
    DP[0][0] = 1
    for x in range(N * M):
        for k in range(1, 7):  #次の目
            y = x + k
            if y >= N * M:
                y = N * M
                for f in range(2):
                    DP[y][f] += DP[x][f]
                    DP[y][f] %= MOD
            else:
                for f in range(2):
                    g = f
                    if y % N == Ci:
                        g = 1
                    DP[y][g] += DP[x][f]
                    DP[y][g] %= MOD
    return DP[-1][-1]


#DPの前計算は必要
#DP[i]: マス0からマスiまでサイコロ移動をしたとき、「ちょうど」マスiに到達する場合の数
DP = [0] * (N * M + 6)
DP[0] = 1
for i in range(N * M):
    for k in range(1, 7):
        DP[i + k] += DP[i]
        DP[i + k] %= MOD

#sub[i]: ゴールが6マス目のとき、iマス目から始めてゴールするまでの場合の数
sub = [0] * 6
for i in range(6):
    dp = [0] * 7
    dp[i] = 1
    for j in range(6):
        for k in range(1, 7):
            dp[min(6, j + k)] += dp[j]
            dp[min(6, j + k)] %= MOD
    sub[i] = dp[-1]
del sub

#包除で解く
#Ciを踏んでゴール + (Ci + N)を踏んでゴール - (CiとCi + Nの両方)を踏んでゴール
def solve(N, M, Ci):
    #1. Ciを踏んでから初期位置にワープし、2 * N - 6 ～ -1に再度移動する
    cnt1 = DP[Ci]
    cnt2 = 0
    for i, j in enumerate(range(M * N - Ci - 6, M * N - Ci), start = 1):
        cnt2 += DP[j] * i  #i通りの出目でゴール可能
        cnt2 %= MOD
    ans = cnt1 * cnt2 % MOD
        
    #2. N + Ciを踏んでから初期位置にワープ
    #3. Ciを踏んでから初期位置にワープ、Nマス先のN + Ciをまた踏む(ダブルカウント)
    cnt1 = DP[N + Ci] - (DP[Ci] * DP[N] % MOD)
    cnt1 %= MOD
    cnt2 = 0
    for i, j in enumerate(range(N - Ci - 6, N - Ci), start = 1):
        if j >= 0:
            cnt2 += DP[j] * i
            cnt2 %= MOD
    ans += cnt1 * cnt2 % MOD
    ans %= MOD
    return ans
    
#答えを出力
for Ci in C:
    print(solve(N, M, Ci))
0