結果

問題 No.2003 Frog on Grid
コンテスト
ユーザー to-omer
提出日時 2022-01-31 18:28:31
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 119 ms / 2,000 ms
コード長 1,300 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 312 ms
コンパイル使用メモリ 85,248 KB
実行使用メモリ 116,016 KB
最終ジャッジ日時 2026-03-16 20:18:56
合計ジャッジ時間 2,743 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

MOD = 998244353


def main():
    H, W, K, C = open(0).read().split(maxsplit=3)
    H, W, K = int(H), int(W), int(K)
    dp = [0] * (H * W)
    stair = [0] * (H * W)
    dacc = [0] * (H * W)
    for i in range(H):
        hacc = [0] * W
        for j in range(W):
            x = 0
            if i == 0 and j == 0:
                dp[0] += 1
            else:
                if i > 0:
                    x += stair[(i - 1) * W + j]
                if j > 0:
                    x += hacc[j - 1]
                if j >= K + 1:
                    x -= hacc[j - K - 1]
                p = max(0, K - j)
                if i >= p + 1:
                    x -= dacc[(i - p - 1) * W + j + p - K]
                if i >= K + 2 and j + 1 < W:
                    x += dacc[(i - K - 2) * W + j + 1]
                x %= MOD
                if C[i * (W + 1) + j] == ".":
                    dp[i * W + j] = x

            stair[i * W + j] = (x + dp[i * W + j]) % MOD
            dacc[i * W + j] = dp[i * W + j]
            if i > 0 and j + 1 < W:
                dacc[i * W + j] = (dacc[i * W + j] + dacc[(i - 1) * W + j + 1]) % MOD
            hacc[j] = dp[i * W + j]
            if j > 0:
                hacc[j] = (hacc[j] + hacc[j - 1]) % MOD

    print(dp[-1])


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