結果

問題 No.1 道のショートカット
ユーザー takakin
提出日時 2020-04-11 22:38:33
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 219 ms / 5,000 ms
コード長 1,084 bytes
コンパイル時間 95 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 22,144 KB
最終ジャッジ日時 2024-07-20 16:46:25
合計ジャッジ時間 4,659 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=lambda: sys.stdin.readline().rstrip()
n=int(input())
c=int(input())
v=int(input())
S=[int(i) for i in input().split()]
T=[int(i) for i in input().split()]
Y=[int(i) for i in input().split()]
M=[int(i) for i in input().split()]
nc=n*(c+1)
edge = [[] for i in range(nc)]
for i in range(v):
  s,t,y,m=S[i],T[i],Y[i],M[i]
  for j in range(y,c+1):
    edge[s-1+j*n].append(m*10**6+t-1+(j-y)*n)

import heapq
def dijkstra_heap(s):
  d=[float("inf")]*nc
  used=[True]*nc
  d[s]=0
  used[s]=False
  edgelist=[]
  for e in edge[s]:
    heapq.heappush(edgelist,e)
  while edgelist:
    minedge=heapq.heappop(edgelist)
    minedge_cost=minedge//(10**6)
    minedge_v=minedge%(10**6)
    if not used[minedge_v]:
      continue
    d[minedge_v] = minedge_cost
    used[minedge_v] = False
    for e in edge[minedge_v]:
      e_cost=e//(10**6)
      e_v=e%(10**6)
      if used[e_v]:
        heapq.heappush(edgelist,e+minedge_cost*10**6)
  return d
D=dijkstra_heap(n*c)
ans=D[n-1]
for i in range(c):
  ans=min(ans,D[n-1+(i+1)*n])
if ans==float("inf"):
  print(-1)
else:
  print(ans)
0