結果

問題 No.1283 Extra Fee
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-01-24 02:11:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,888 ms / 2,000 ms
コード長 1,817 bytes
コンパイル時間 355 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 105,404 KB
最終ジャッジ日時 2024-01-24 02:11:51
合計ジャッジ時間 21,640 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
53,460 KB
testcase_01 AC 39 ms
53,460 KB
testcase_02 AC 39 ms
53,460 KB
testcase_03 AC 37 ms
53,460 KB
testcase_04 AC 39 ms
53,460 KB
testcase_05 AC 38 ms
53,460 KB
testcase_06 AC 41 ms
53,460 KB
testcase_07 AC 39 ms
53,460 KB
testcase_08 AC 43 ms
53,460 KB
testcase_09 AC 40 ms
53,460 KB
testcase_10 AC 40 ms
53,460 KB
testcase_11 AC 235 ms
78,600 KB
testcase_12 AC 247 ms
78,280 KB
testcase_13 AC 198 ms
77,412 KB
testcase_14 AC 316 ms
79,348 KB
testcase_15 AC 446 ms
81,312 KB
testcase_16 AC 190 ms
77,560 KB
testcase_17 AC 1,760 ms
103,792 KB
testcase_18 AC 1,321 ms
94,092 KB
testcase_19 AC 1,321 ms
93,164 KB
testcase_20 AC 1,202 ms
91,888 KB
testcase_21 AC 1,245 ms
92,704 KB
testcase_22 AC 1,106 ms
90,352 KB
testcase_23 AC 1,139 ms
89,076 KB
testcase_24 AC 1,156 ms
89,204 KB
testcase_25 AC 1,366 ms
94,216 KB
testcase_26 AC 1,335 ms
94,312 KB
testcase_27 AC 1,305 ms
93,648 KB
testcase_28 AC 1,342 ms
94,044 KB
testcase_29 AC 1,888 ms
105,404 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

import heapq

MAX_INT = 10 ** 18

def main():
    N, M = map(int, input().split())
    pass_map = [[0] * N for _ in range(N)]
    for _ in range(M):
        h, w, c = map(int, input().split())
        pass_map[h - 1][w - 1] = c
    
    seen = [[[MAX_INT] * N for _ in range(N)] for _ in range(2)]
    seen[0][0][0] = 0
    fix = [[[MAX_INT] * N for _ in range(N)] for _ in range(2)]
    queue = []
    heapq.heappush(queue, (0, 0, 0, 0))
    while len(queue) > 0:
        cost, state, h, w = heapq.heappop(queue) 
        if fix[state][h][w] <= cost:
            continue

        fix[state][h][w] = cost
        for dh, dw in ((0, 1), (1, 0), (0, -1), (-1, 0)):
            new_h = h + dh
            new_w = w + dw
            if 0 <= new_h and new_h < N and 0 <= new_w and new_w < N:
                # 普通に手数料を払う場合
                if fix[state][new_h][new_w] >= MAX_INT:
                    new_cost = cost + 1 + pass_map[new_h][new_w]
                    if seen[state][new_h][new_w] > new_cost:
                        seen[state][new_h][new_w] = new_cost
                        heapq.heappush(queue, (new_cost, state, new_h, new_w))
                
                # 免除を使う場合
                if state == 0 and pass_map[new_h][new_w] > 0:
                    if fix[1][new_h][new_w] >= MAX_INT:
                        new_cost = cost + 1
                        if seen[1][new_h][new_w] > new_cost:
                            seen[1][new_h][new_w] = new_cost
                            heapq.heappush(queue, (new_cost, 1, new_h, new_w))
    
    answer = min(fix[0][N - 1][N - 1], fix[1][N - 1][N - 1])
    if answer == float("inf"):
        print(-1)
    else:
        print(answer)









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