結果
| 問題 |
No.1320 Two Type Min Cost Cycle
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 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 |
ソースコード
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")