結果

問題 No.1283 Extra Fee
ユーザー LyricalMaestro
提出日時 2024-01-24 02:03:02
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,844 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 82,120 KB
実行使用メモリ 171,468 KB
最終ジャッジ日時 2024-09-28 06:56:49
合計ジャッジ時間 30,552 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

import heapq

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 = [[[float("inf"), float("inf")] for _ in range(N)] for _ in range(N)]
    seen[0][0][0] = 0
    fix = [[[float("inf"), float("inf")] for _ in range(N)] for _ in range(N)]
    queue = []
    heapq.heappush(queue, (0, 0, 0, 0))
    while len(queue) > 0:
        cost, h, w, state = heapq.heappop(queue) 
        if fix[h][w][state] <= cost:
            continue

        fix[h][w][state] = 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[new_h][new_w][state] != float("inf"):
                    continue

                new_cost = cost + 1 + pass_map[new_h][new_w]
                if seen[new_h][new_w][state] > new_cost:
                    seen[new_h][new_w][state] = new_cost
                    heapq.heappush(queue, (new_cost, new_h, new_w, state))
                
                # 免除を使う場合
                if state == 0 and pass_map[new_h][new_w] > 0:
                    if fix[new_h][new_w][1] != float("inf"):
                        continue

                    new_cost = cost + 1
                    if seen[new_h][new_w][1] > new_cost:
                        seen[new_h][new_w][1] = new_cost
                        heapq.heappush(queue, (new_cost, new_h, new_w, 1))
    
    answer = min(fix[N - 1][N - 1])
    if answer == float("inf"):
        print(-1)
    else:
        print(answer)









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