結果

問題 No.2003 Frog on Grid
ユーザー to-omer
提出日時 2022-01-31 18:28:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 187 ms / 2,000 ms
コード長 1,300 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 111,504 KB
最終ジャッジ日時 2024-06-28 08:09:59
合計ジャッジ時間 3,802 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

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