結果
| 問題 | 
                            No.3111 Toll Optimization
                             | 
                    
| コンテスト | |
| ユーザー | 
                             nikoro256
                         | 
                    
| 提出日時 | 2025-04-18 22:08:25 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 1,034 ms / 5,000 ms | 
| コード長 | 890 bytes | 
| コンパイル時間 | 585 ms | 
| コンパイル使用メモリ | 82,460 KB | 
| 実行使用メモリ | 172,220 KB | 
| 最終ジャッジ日時 | 2025-04-18 22:08:59 | 
| 合計ジャッジ時間 | 30,339 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge3 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 70 | 
ソースコード
import heapq
inf = 10**17
def dijkstra(s):
    hq=[]
    heapq.heappush(hq,(0,s))
    dist=[inf for _ in range(N*(K+1))]
    while len(hq)!=0:
        d,p=heapq.heappop(hq)
        if dist[p] < d:
            continue
        dist[p]=d
        for e,c in edge[p]:
            if dist[e]>d+c:
                dist[e]=d+c
                heapq.heappush(hq,(d+c,e))
    return dist
N,M,K=map(int,input().split())
C=list(map(int,input().split()))
edge=[[] for _ in range(N*(K+1))]
for j in range(M):
    u,v=map(int,input().split())
    u-=1
    v-=1
    for i in range(K+1):
        edge[i*N+u].append((i*N+v,C[j]))
        edge[i*N+v].append((i*N+u,C[j]))
    for i in range(K):
        edge[i*N+u].append(((i+1)*N+v,0))
        edge[i*N+v].append(((i+1)*N+u,0))
dist = dijkstra(0)
ans=inf
for i in range(K+1):
    ans=min(ans,dist[i*N+N-1])
if ans==inf:
    print(-1)
else:
    print(ans)
            
            
            
        
            
nikoro256