結果
問題 | No.1545 [Cherry 2nd Tune N] Anthem |
ユーザー |
|
提出日時 | 2022-11-09 21:50:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,227 ms / 3,000 ms |
コード長 | 1,276 bytes |
コンパイル時間 | 171 ms |
コンパイル使用メモリ | 82,456 KB |
実行使用メモリ | 451,348 KB |
最終ジャッジ日時 | 2024-07-23 02:57:23 |
合計ジャッジ時間 | 39,034 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 67 |
ソースコード
import heapqINF = 1 << 60n, s, t, k = map(int, input().split())s -= 1t -= 1x = list(map(int, input().split()))m = int(input())edges = []idx = {}vs = [s, (k - 1) * n + t]idx[vs[0]] = 0idx[vs[1]] = 1src = 0dst = 1for _ in range(m):a, b, y = map(int, input().split())a -= 1b -= 1for i in range(k):u = i * n + av = min(i + 1, k - 1) * n + bif idx.get(u, None) is None:idx[u] = len(idx)vs.append(u)if idx.get(v, None) is None:idx[v] = len(idx)vs.append(v)edges.append((idx[u], idx[v], y + x[b]))g = [[] for _ in range(len(idx))]for u, v, w in edges:g[u].append((v, w))dist = [INF for _ in range(len(idx))]dist[src] = x[s]prev = [None for _ in range(len(idx))]hp = [(dist[src], src)]while len(hp) > 0:cd, cur = heapq.heappop(hp)if cd > dist[cur]:continuefor nxt, cost in g[cur]:if dist[cur] + cost < dist[nxt]:dist[nxt] = dist[cur] + costprev[nxt] = curheapq.heappush(hp, (dist[nxt], nxt))if dist[dst] < INF:print("Possible")print(dist[dst])cur = dstorder = []while not cur is None:order.append(vs[cur] % n + 1)cur = prev[cur]order.reverse()print(len(order))print(*order, sep=" ")else:print("Impossible")