結果

問題 No.807 umg tours
コンテスト
ユーザー TakoKurage
提出日時 2019-03-22 22:37:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,681 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 141,428 KB
最終ジャッジ日時 2024-09-19 06:03:57
合計ジャッジ時間 12,482 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 2 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from heapq import heappop, heappush


class Graph(object):
    def __init__(self):
        self.graph = defaultdict(list)

    def __len__(self):
        return len(self.graph)

    def add_edge(self, src, dst, weight=1):
        self.graph[src].append((dst, weight))

    def get_nodes(self):
        return self.graph.keys()


class Dijkstra(object):
    def __init__(self, graph, start):
        self.g = graph.graph
        self.dist = defaultdict(lambda: float('inf'))
        self.dist[start] = 0
        self.prev = defaultdict(lambda: None)
        self.Q = []
        heappush(self.Q, (self.dist[start], start))
        while self.Q:
            dist_u, u = heappop(self.Q)
            if self.dist[u] < dist_u:
                continue
            for v, weight in self.g[u]:
                alt = dist_u + weight
                if self.dist[v] > alt:
                    self.dist[v] = alt
                    self.prev[v] = u
                    heappush(self.Q, (alt, v))

    def shortest_distance(self, goal):
        return self.dist[goal]


N, M = [int(i) for i in input().split()]
graphs = [Graph() for _ in range(M + 1)]
inputs = [(int(i) for i in input().split()) for _ in range(M)]

for i, (a, b, c) in enumerate(inputs):
    for j, graph in enumerate(graphs):
        if i == j:
            graph.add_edge(a, b, 0)
            graph.add_edge(b, a, 0)
        else:
            graph.add_edge(a, b, c)
            graph.add_edge(b, a, c)

for j in range(N):
    result = Dijkstra(graphs[0], 1).shortest_distance(
        j + 1) + min([Dijkstra(graph, 1).shortest_distance(j + 1) for graph in graphs[1:]])
    print(result)
0