結果

問題 No.1283 Extra Fee
ユーザー mkawa2
提出日時 2020-11-07 11:44:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,371 ms / 2,000 ms
コード長 863 bytes
コンパイル時間 228 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 120,392 KB
最終ジャッジ日時 2024-11-16 07:04:35
合計ジャッジ時間 18,128 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input = sys.stdin.buffer.readline

def MI(): return map(int, input().split())
dij = [(1, 0), (0, 1), (-1, 0), (0, -1)]

from heapq import *

inf = 10**16
n, m = MI()
cc = [[0]*n for _ in range(n)]
for _ in range(m):
    i, j, c = MI()
    cc[i-1][j-1] = c

hp = []
dd = [[[inf]*2 for _ in range(n)] for _ in range(n)]
dd[0][0][0] = 0
heappush(hp, (0, 0, 0, 0))
while hp:
    d, i, j, k = heappop(hp)
    if (i,j)==(n-1,n-1):
        print(d)
        exit()
    if d > dd[i][j][k]: continue
    for di, dj in dij:
        ni, nj = i+di, j+dj
        if ni < 0 or ni >= n or nj < 0 or nj >= n: continue
        nd = d+cc[ni][nj]+1
        if nd < dd[ni][nj][k]:
            dd[ni][nj][k] = nd
            heappush(hp, (nd, ni, nj, k))
        if k == 0 and d+1 < dd[ni][nj][1]:
            dd[ni][nj][1] = d+1
            heappush(hp, (d+1, ni, nj, 1))
0