結果
問題 |
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")