結果

問題 No.2855 Move on Grid
ユーザー ルクルク
提出日時 2024-08-25 16:28:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,566 bytes
コンパイル時間 234 ms
コンパイル使用メモリ 82,300 KB
実行使用メモリ 328,260 KB
最終ジャッジ日時 2024-08-25 16:28:54
合計ジャッジ時間 21,150 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
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 AC 569 ms
327,468 KB
testcase_11 AC 540 ms
327,740 KB
testcase_12 AC 574 ms
327,340 KB
testcase_13 AC 540 ms
327,300 KB
testcase_14 AC 577 ms
327,156 KB
testcase_15 AC 524 ms
327,736 KB
testcase_16 AC 555 ms
327,356 KB
testcase_17 AC 530 ms
327,348 KB
testcase_18 AC 526 ms
327,264 KB
testcase_19 AC 549 ms
327,328 KB
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 WA -
testcase_38 WA -
testcase_39 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import itertools
from heapq import *
from functools import lru_cache

# 定数の設定
SMALLMOD = 998244353
BIGMOD = 1000000007
PI = 3.141592653589793
ALPHABET = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
alphabet = "abcdefghijklmnopqrstuvwxyz"
dx = [1, 0, -1, 0, 1, -1, -1, 1]
dy = [0, 1, 0, -1, 1, 1, -1, -1]

def main():
    # 入力の受け取り
    n, m, k = map(int, sys.stdin.readline().split())
    mp = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
    
    # dp配列の初期化
    dp = [[[] for _ in range(m)] for _ in range(n)]
    dp[0][0] = [mp[0][0]]

    # 最初の列のdpを更新
    for i in range(1, n):
        tmp = dp[i - 1][0] + [mp[i][0]]
        tmp.sort()
        dp[i][0] = tmp

    # 最初の行のdpを更新
    for j in range(1, m):
        tmp = dp[0][j - 1] + [mp[0][j]]
        tmp.sort()
        dp[0][j] = tmp

    # dp配列の更新
    for i in range(1, n):
        for j in range(1, m):
            tmp1 = dp[i - 1][j] + [mp[i][j]]
            tmp2 = dp[i][j - 1] + [mp[i][j]]
            tmp1.sort()
            tmp2.sort()
            t1 = min(tmp1[k-1:]) if k <= len(tmp1) else float('inf')
            t2 = min(tmp2[k-1:]) if k <= len(tmp2) else float('inf')
            dp[i][j] = tmp1 if t1 > t2 else tmp2

    # 最後のdpの要素を取得して、最小値を出力
    last = dp[-1][-1]
    last.sort()
    for i in range(k):
        if i < len(last):
            last[i] = float('inf')
    print(1000000000 if min(last)==float("inf") else min(last))

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