結果

問題 No.1320 Two Type Min Cost Cycle
ユーザー qib
提出日時 2023-04-09 00:08:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,066 ms / 2,000 ms
コード長 896 bytes
コンパイル時間 155 ms
コンパイル使用メモリ 82,564 KB
実行使用メモリ 82,804 KB
最終ジャッジ日時 2024-10-03 21:36:35
合計ジャッジ時間 16,002 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 57
権限があれば一括ダウンロードができます

ソースコード

diff #

import heapq

INF = 1 << 60

t = int(input())
n, m = map(int, input().split())
edges = []
for _ in range(m):
  u, v, w = map(int, input().split())
  edges.append((u - 1, v - 1, w))

g = [[] for _ in range(n)]
if t == 0:
  for i in range(m):
    u, v, w = edges[i]
    g[u].append((v, w, i))
    g[v].append((u, w, i))
else:
  for i in range(m):
    u, v, w = edges[i]
    g[u].append((v, w, i))

ans = INF
for i in range(m):
  dst, src, w = edges[i]
  dist = [INF for _ in range(n)]
  dist[src] = 0
  hp = [(0, src)]
  while len(hp) > 0:
    cd, cur = heapq.heappop(hp)
    if cur == dst:
      break
    if cd > dist[cur]:
      continue
    for nxt, cost, j in g[cur]:
      if j == i or dist[nxt] <= dist[cur] + cost:
        continue
      dist[nxt] = dist[cur] + cost
      heapq.heappush(hp, (dist[nxt], nxt))

  ans = min(ans, dist[dst] + w)

if ans < INF:
  print(ans)
else:
  print("-1")
0