結果

問題 No.160 最短経路のうち辞書順最小
ユーザー はむ吉🐹
提出日時 2016-07-18 15:09:39
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 115 ms / 5,000 ms
コード長 2,089 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 15,488 KB
最終ジャッジ日時 2024-10-15 16:53:22
合計ジャッジ時間 2,734 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3

import array
import collections
import heapq


INF = 10 ** 9
UNDEF = -1


# ゴールからスタートまでの最短経路をDijkstra法で求め、逆からたどる
def compute_optimal_path(start, goal, num_vertices, adj_edges):
    # ゴールから各頂点への最短距離を格納する
    dist = array.array("L", (INF for _ in range(num_vertices)))
    dist[goal] = 0
    # 幅優先探索順序で「ひとつ前」の頂点を格納する
    pred = array.array("i", (UNDEF for _ in range(num_vertices)))
    # 優先度付きキュー(二分ヒープ)に最短距離と頂点の組を入れる
    pq = [(dist[v], v) for v in range(num_vertices)]
    heapq.heapify(pq)
    while pq:
        _, v = heapq.heappop(pq)
        if v == start:
            # 取り出してきた頂点がスタートなら終わる
            break
        for u, cost in adj_edges[v]:
            new_length = dist[v] + cost
            if new_length < dist[u]:
                dist[u] = new_length
                pred[u] = v
                heapq.heappush(pq, (new_length, u))
            elif new_length == dist[u]:
                # 距離が同じなら、pred[]を辞書順でより小さい頂点に置き換える
                pred[u] = min(pred[u], v)
    # ここから経路復元
    # 後ろからpred[]をたどっていくと、それが求める経路になる
    path = [start]
    current = start
    while pred[current] != UNDEF:
        current = pred[current]
        path.append(current)
    return path


def main():
    num_vertices, num_edges, start, goal = map(int, input().split())
    adj_edges = [set() for _ in range(num_vertices)]
    for _ in range(num_edges):
        a, b, c = map(int, input().split())
        # 無向グラフ表すのため両向きの辺を考える
        # (行先, コスト)の組で隣接する辺を表す
        adj_edges[a].add((b, c))
        adj_edges[b].add((a, c))
    path = compute_optimal_path(start, goal, num_vertices, adj_edges)
    print(*path)


if __name__ == '__main__':
    main()
0