結果

問題 No.971 いたずらっ子
ユーザー ttr
提出日時 2020-01-29 20:21:58
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,548 bytes
コンパイル時間 232 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 158,212 KB
最終ジャッジ日時 2024-09-16 02:07:25
合計ジャッジ時間 7,035 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample -- * 4
other TLE * 1 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

import heapq
H,W = map(int, input().split())
L = [input() for _ in range(H)]

inf = 10**15
dist = [[inf]*W for _ in range(H)]
h = []
heapq.heappush(h, (0, 0, 0, 0))
while dist[H-1][W-1] == inf:
    temp = heapq.heappop(h)
    d = temp[0]
    i = temp[1]
    j = temp[2]
    sub_d = temp[3]
    if i > 0:
        if L[i-1][j] == "." and dist[i-1][j] > d+1:
            dist[i-1][j] = d+1
            heapq.heappush(h, (d+1, i-1, j, sub_d+1))
        elif L[i-1][j] == "k" and dist[i-1][j] > d+2+sub_d:
            dist[i-1][j] = d+2+sub_d
            heapq.heappush(h, (d+2+sub_d, i-1, j, sub_d+1))
    if i < H-1:
        if L[i+1][j] == "." and dist[i+1][j] > d+1:
            dist[i+1][j] = d+1
            heapq.heappush(h, (d+1, i+1, j, sub_d+1))
        elif L[i+1][j] == "k" and dist[i+1][j] > d+2+sub_d:
            dist[i+1][j] = d+2+sub_d
            heapq.heappush(h, (d+2+sub_d, i+1, j, sub_d+1))
    if j > 0:
        if L[i][j-1] == "." and dist[i][j-1] > d+1:
            dist[i][j-1] = d+1
            heapq.heappush(h, (d+1, i, j-1, sub_d+1))
        elif L[i][j-1] == "k" and dist[i][j-1] > d+2+sub_d:
            dist[i][j-1] = d+2+sub_d
            heapq.heappush(h, (d+2+sub_d, i, j-1, sub_d+1))
    if j < W-1:
        if L[i][j+1] == "." and dist[i][j+1] > d+1:
            dist[i][j+1] = d+1
            heapq.heappush(h, (d+1, i, j+1, sub_d+1))
        elif L[i][j+1] == "k" and dist[i][j+1] > d+2+sub_d:
            dist[i][j+1] = d+2+sub_d
            heapq.heappush(h, (d+2+sub_d, i, j+1, sub_d+1))

print(dist[H-1][W-1])
0