結果

問題 No.867 避難経路
ユーザー vwxyz
提出日時 2024-04-14 17:01:33
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,017 bytes
コンパイル時間 291 ms
コンパイル使用メモリ 82,272 KB
実行使用メモリ 232,980 KB
最終ジャッジ日時 2024-10-03 17:42:43
合計ジャッジ時間 118,779 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

H,W=map(int,input().split())
gx,gy=map(int,input().split())
gx-=1;gy-=1
A=[list(map(int,input().split())) for h in range(H)]
Q=int(input())
K=51
query=[[] for k in range(K+1)]
for q in range(Q):
    x,y,k=map(int,input().split())
    x-=1;y-=1
    query[min(k,K)].append((x,y,k,q))
ans_lst=[None]*Q
for k in range(K+1):
    w=k**2 if k<K else 1<<30
    inf=1<<60
    dist=[[inf]*W for h in range(H)]
    dist[gx][gy]=0
    queue=[(dist[gx][gy],gx,gy)]
    while queue:
        d,x,y=heapq.heappop(queue)
        for dx,dy in ((0,1),(1,0),(0,-1),(-1,0)):
            if 0<=x+dx<H and 0<=y+dy<W:
                xx,yy=x+dx,y+dy
                dd=d+w+A[xx][yy]
                if dist[xx][yy]>dd:
                    dist[xx][yy]=dd
                    heapq.heappush(queue,(dd,xx,yy))
    for x,y,c,q in query[k]:
        if k<K:
            ans_lst[q]=dist[x][y]+c**2+A[gx][gy]
        else:
            cnt,d=divmod(dist[x][y],w)
            ans_lst[q]=(cnt+1)*c**2+d+A[gx][gy]
print(*ans_lst,sep="\n")
0