結果

問題 No.1814 Uribo Road (Easy)
ユーザー rin204
提出日時 2022-01-20 22:36:13
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 186 ms / 2,000 ms
コード長 828 bytes
コンパイル時間 133 ms
コンパイル使用メモリ 82,372 KB
実行使用メモリ 82,304 KB
最終ジャッジ日時 2024-11-24 08:12:06
合計ジャッジ時間 6,751 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import *

n, m, k = map(int, input().split())
R = list(map(int, input().split()))
R = {r - 1:1 << i for i, r in enumerate(R)}
edges = [[] for _ in range(n)]
for i in range(m):
    a, b, c = map(int, input().split())
    a -= 1
    b -= 1
    if i in R:
        d = R[i]
    else:
        d = -1
    edges[a].append((b, c, d))
    edges[b].append((a, c, d))
    
inf = 1 << 40
dist = [[inf] * (1 << k) for _ in range(n)]
dist[0][0] = 0
hq = [(0, 0, 0)]
while hq:
    d, pos, bit = heappop(hq)
    if d > dist[pos][bit]:
        continue
    
    for npos, c, dbit in edges[pos]:
        if dbit == -1:
            nbit = bit
        else:
            nbit = bit | dbit
        
        if dist[npos][nbit] > c + d:
            dist[npos][nbit] = c + d
            heappush(hq, (c + d, npos, nbit))
print(dist[-1][-1])
0