tests/dijkstra_getpath.py

View this file on GitHub
Last update: 2025-09-11 18:15:20+09:00
Problem: https://judge.yosupo.jp/problem/shortest_path

Depends on

Code

# competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/shortest_path
from libs.dijkstra import dijkstra
from libs.get_path import getpath
from libs.graph import GraphW
from libs.standard_input import il
from libs.utils import INF

N, M, S, T = il()
G = GraphW(N, dire=True)

for _ in [0] * M:
    a, b, w = il()
    G.new_side(a, b, w)

used, prev = dijkstra(G.all(), S, True)

if used[T] == INF:
    print(-1)
else:
    path = getpath(prev, T)

    print(used[T], len(path) - 1)

    for i in range(len(path) - 1):
        print(path[i], path[i + 1])

Test cases

Env	Name	Status	Elapsed	Memory
Python	almost_line_00	AC	1891 ms	118 MB
Python	almost_line_01	AC	1877 ms	118 MB
Python	almost_line_02	AC	1827 ms	98 MB
Python	example_00	AC	24 ms	12 MB
Python	example_01	AC	25 ms	12 MB
Python	grid_random_00	AC	1581 ms	94 MB
Python	grid_swirl_00	AC	1904 ms	102 MB
Python	line_00	AC	2497 ms	165 MB
Python	max_dense_long_00	AC	1077 ms	78 MB
Python	max_dense_random_00	AC	1063 ms	78 MB
Python	max_dense_random_01	AC	1084 ms	78 MB
Python	max_dense_zero_00	AC	1018 ms	62 MB
Python	max_sparse_random_00	AC	1390 ms	103 MB
Python	max_sparse_random_01	AC	1911 ms	116 MB
Python	max_sparse_random_02	AC	1921 ms	116 MB
Python	max_star_00	AC	2615 ms	177 MB
Python	max_star_01	AC	2750 ms	161 MB
Python	small_00	AC	25 ms	12 MB
Python	small_01	AC	25 ms	12 MB
Python	small_02	AC	25 ms	12 MB
Python	small_03	AC	25 ms	12 MB
Python	small_04	AC	25 ms	12 MB
Python	sparse_random_00	AC	1206 ms	105 MB
Python	sparse_random_01	AC	1249 ms	115 MB
Python	sparse_random_02	AC	1256 ms	76 MB
Python	spfa_killer_00	AC	2293 ms	122 MB
Python	wrong_dijkstra_handmade_00	AC	25 ms	12 MB
Python	wrong_dijkstra_killer_00	AC	2132 ms	136 MB
Python	wrong_dijkstra_killer_01	AC	2437 ms	124 MB