結果

問題 No.1283 Extra Fee
ユーザー wgrape
提出日時 2024-11-14 17:21:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,947 ms / 2,000 ms
コード長 1,365 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 82,208 KB
実行使用メモリ 142,744 KB
最終ジャッジ日時 2024-11-14 17:22:14
合計ジャッジ時間 26,176 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M = map(int,input().split())
G = [[1] * N for i in range(N)]
G[0][0] = 0

for _ in range(M):
    h,w,c = map(int,input().split())
    G[h - 1][w - 1] += c
    
import heapq as hq
q = []
hq.heappush(q, (0, 0, 0, 0))
visited = [[[False] * 2 for i in range(N)] for j in range(N)]
INF = 1 << 60
best = [[[INF] * 2 for i in range(N)] for j in range(N)]
cases = ((1,0),(0,1),(-1,0),(0,-1))
while q:
    c, used, i, j = hq.heappop(q)
    if visited[i][j][used]:
        continue
    visited[i][j][used] = True
    if i == N - 1 and j == N - 1:
        print(c)
        break
    for ci,cj in cases:
        ni = i + ci
        nj = j + cj
        if not (0 <= ni < N and 0 <= nj < N):
            continue
        if G[ni][nj] > 1 and used == 0: # まだ権利を使っていない
            if not visited[ni][nj][used + 1] and best[ni][nj][used + 1] > c + 1:
                hq.heappush(q, (c + 1, used + 1, ni, nj))
                best[ni][nj][used + 1] = c + 1
            if not visited[ni][nj][used] and best[ni][nj][used] > c + G[ni][nj]:
                hq.heappush(q, (c + G[ni][nj], used, ni, nj))
                best[ni][nj][used] = c + G[ni][nj]
        else:
            if not visited[ni][nj][used] and best[ni][nj][used] > c + G[ni][nj]:
                hq.heappush(q, (c + G[ni][nj], used, ni, nj))
                best[ni][nj][used] = c + G[ni][nj]
0