結果

問題 No.2673 A present from B
ユーザー 👑 p-adic
提出日時 2024-06-15 15:20:13
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 906 ms / 2,000 ms
コード長 427 bytes
コンパイル時間 549 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 176,136 KB
最終ジャッジ日時 2024-06-15 15:20:24
合計ジャッジ時間 8,011 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

R=range
J=lambda:map(int,input().split())
N,M=J()
A=list(J())
P=[]
V=N*M+N
E=[]
for v in R(V):
	i,j,e=v//N,v%N,[]
	if i:a,n=A[i-1],i*N-N;e+=[[n+(a if j==a-1else a-1if j==a else j),0]]
	if j:e+=[[i*N+j-1,1]]
	if j+1<N:e+=[[i*N+j+1,1]]
	E+=[e]
import heapq
W=[9**20]*V
W[M*N]=0
Q=[[0,M*N]]
while len(Q):
	w,i=heapq.heappop(Q)
	if w>W[i]:continue
	for[j,v]in E[i]:
		u=w+v
		if u<W[j]:W[j]=u;heapq.heappush(Q,[u,j])
print(*W[1:N])
0