結果

問題 No.1283 Extra Fee
コンテスト
ユーザー LyricalMaestro
提出日時 2024-01-24 02:11:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,925 ms / 2,000 ms
コード長 1,817 bytes
コンパイル時間 355 ms
コンパイル使用メモリ 82,196 KB
実行使用メモリ 106,024 KB
最終ジャッジ日時 2024-09-28 06:57:13
合計ジャッジ時間 21,716 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

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