結果

問題 No.1812 Uribo Road
ユーザー tobusakanatobusakana
提出日時 2022-10-23 20:15:10
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,440 bytes
コンパイル時間 1,197 ms
コンパイル使用メモリ 86,984 KB
実行使用メモリ 89,592 KB
最終ジャッジ日時 2023-09-15 05:43:24
合計ジャッジ時間 12,262 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 76 ms
71,260 KB
testcase_01 AC 74 ms
71,380 KB
testcase_02 AC 75 ms
71,480 KB
testcase_03 AC 263 ms
81,176 KB
testcase_04 AC 74 ms
71,340 KB
testcase_05 AC 74 ms
71,116 KB
testcase_06 AC 75 ms
71,448 KB
testcase_07 AC 81 ms
75,620 KB
testcase_08 AC 428 ms
83,408 KB
testcase_09 AC 733 ms
89,592 KB
testcase_10 AC 333 ms
82,016 KB
testcase_11 AC 224 ms
80,328 KB
testcase_12 TLE -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline = sys.stdin.readline

# dp[v][S] = 今までに触った猪の集合がSでvにいるときの最小コスト

N,M,K = map(int,readline().split())
R = list(map(int,readline().split()))
Rdic = {} # 道の番号から猪の番号を導く
for i in range(K):
  Rdic[R[i] - 1] = i
  
G = [[] for i in range(N)]
for i in range(M):
  a,b,c = map(int,readline().split())
  ino = -1
  if i in Rdic: # 道の番号が含まれる
    ino = Rdic[i] # イノシシの番号に変換
  G[a - 1].append([b - 1, c, ino])
  G[b - 1].append([a - 1, c, ino])
  
INF = 1 << 60
dp = [[INF] * (1 << K) for i in range(N)]
best = [[INF] * (1 << K) for i in range(N)]
import heapq as hq
q = []
hq.heappush(q, (0, 0, 0)) # 距離、頂点、今までに触った猪の集合
complete = (1 << K) - 1
while q:
  d,v,status = hq.heappop(q)
  if dp[v][status] != INF:
    continue
  dp[v][status] = d
  if v == N - 1 and status == complete:
    print(dp[v][status])
    break
  for child, cost, ino in G[v]:
    if ino != -1:
      next_status = status | (1 << ino)
      if dp[child][next_status] != INF or best[child][next_status] <= d + cost:
        continue
      best[child][next_status] = d + cost
      hq.heappush(q, (d + cost, child, status | (1 << ino)))
    else:
      if dp[child][status] != INF or best[child][status] <= d + cost:
        continue
      best[child][status] = d + cost
      hq.heappush(q, (d + cost, child, status))
0