結果
問題 |
No.807 umg tours
|
ユーザー |
|
提出日時 | 2022-06-10 02:32:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,632 ms / 4,000 ms |
コード長 | 2,557 bytes |
コンパイル時間 | 186 ms |
コンパイル使用メモリ | 82,568 KB |
実行使用メモリ | 327,308 KB |
最終ジャッジ日時 | 2024-09-21 05:41:20 |
合計ジャッジ時間 | 23,967 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
# import pypyjit # pypyjit.set_param('max_unroll_recursion=-1') import sys from itertools import combinations, permutations, product, accumulate, groupby from collections import defaultdict, deque, Counter from functools import reduce from operator import add, mul import heapq import bisect import math import copy sys.setrecursionlimit(10 ** 9) input = lambda: sys.stdin.readline().rstrip() INF = float("inf") MOD = 10 ** 9 + 7 import collections import heapq class Dijkstra(): def __init__(self): self.e = collections.defaultdict(list) def add(self, u, v, d, directed=False): """ #0-indexedでなくてもよいことに注意 #u = from, v = to, d = cost #directed = Trueなら、有向グラフである """ if directed is False: self.e[u].append([v, d]) self.e[v].append([u, d]) else: self.e[u].append([v, d]) def delete(self, u, v): self.e[u] = [_ for _ in self.e[u] if _[0] != v] self.e[v] = [_ for _ in self.e[v] if _[0] != u] def Dijkstra_search(self, s): """ #0-indexedでなくてもよいことに注意 #:param s: 始点 #:return: 始点から各点までの最短経路と最短経路を求めるのに必要なprev """ d = collections.defaultdict(lambda: float('inf')) prev = collections.defaultdict(lambda: None) d[s] = 0 q = [] heapq.heappush(q, (0, s)) v = collections.defaultdict(bool) while len(q): k, u = heapq.heappop(q) if v[u]: continue v[u] = True for uv, ud in self.e[u]: if v[uv]: continue vd = k + ud if d[uv] > vd: d[uv] = vd prev[uv] = u heapq.heappush(q, (vd, uv)) return d, prev def getDijkstraShortestPath(self, start, goal): _, prev = self.Dijkstra_search(start) shortestPath = [] node = goal while node is not None: shortestPath.append(node) node = prev[node] return shortestPath[::-1] N, M = map(int, input().split()) ABC = [list(map(int, input().split())) for i in range(M)] graph = Dijkstra() for a, b, c in ABC: graph.add(a, b, c) graph.add(a, -b, 0, True) graph.add(b, -a, 0, True) graph.add(-a, -b, c) g, _ = graph.Dijkstra_search(1) print(0) for i in range(2, N + 1): print(g[i] + g[-i])