結果

問題 No.1283 Extra Fee
ユーザー tanon710tanon710
提出日時 2020-11-07 01:40:59
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,445 bytes
コンパイル時間 341 ms
コンパイル使用メモリ 82,128 KB
実行使用メモリ 156,792 KB
最終ジャッジ日時 2024-07-22 14:02:55
合計ジャッジ時間 27,383 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 23 TLE * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

n,m=map(int,input().split())
fees={}
for _ in range(m):
    y,x,cost=map(int,input().split())
    fees[(x-1,y-1)]=cost
INF=10**18
costs=[[[INF,INF] for _ in range(n)] for _ in range(n)]
costs[0][0]=[0,0]
q=[]
heapq.heappush(q,(0,0,0,0))
while len(q)!=0:
    curcost,x,y,skipped=heapq.heappop(q)
    if x==n-1 and y==n-1:
        print(curcost)
        exit()
    for dx,dy in [(-1,0),(1,0),(0,-1),(0,1)]:
        if 0<=x+dx<=n-1 and 0<=y+dy<=n-1:
            if (x+dx,y+dy) in fees:
                if skipped==0:
                    if costs[y+dy][x+dx][skipped+1]>curcost+1:
                        heapq.heappush(q,(curcost+1,x+dx,y+dy,skipped+1))
                        costs[y+dy][x+dx][skipped+1]=curcost+1
                    if costs[y+dy][x+dx][skipped]>curcost+1+fees[(x+dx,y+dy)]:
                        heapq.heappush(q,(curcost+1+fees[(x+dx,y+dy)],x+dx,y+dy,skipped))
                        costs[y+dy][x+dx][skipped]=curcost+1+fees[(x+dx,y+dy)]
                else:
                    if costs[y+dy][x+dx][skipped]>curcost+1+fees[(x+dx,y+dy)]:
                        heapq.heappush(q,(curcost+1+fees[(x+dx,y+dy)],x+dx,y+dy,skipped))
                        costs[y+dy][x+dx][skipped]=curcost+1+fees[(x+dx,y+dy)]
            else:
                if costs[y+dy][x+dx][skipped]>curcost+1:
                    heapq.heappush(q,(curcost+1,x+dx,y+dy,skipped))
                    costs[y+dy][x+dx][skipped]=curcost+1
0