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	1965 ms	118 MB
Python	almost_line_01	AC	1963 ms	118 MB
Python	almost_line_02	AC	1857 ms	98 MB
Python	example_00	AC	30 ms	12 MB
Python	example_01	AC	29 ms	12 MB
Python	grid_random_00	AC	1680 ms	94 MB
Python	grid_swirl_00	AC	1977 ms	102 MB
Python	line_00	AC	2480 ms	165 MB
Python	max_dense_long_00	AC	1171 ms	77 MB
Python	max_dense_random_00	AC	1170 ms	77 MB
Python	max_dense_random_01	AC	1252 ms	77 MB
Python	max_dense_zero_00	AC	1130 ms	61 MB
Python	max_sparse_random_00	AC	1515 ms	103 MB
Python	max_sparse_random_01	AC	2024 ms	116 MB
Python	max_sparse_random_02	AC	2022 ms	116 MB
Python	max_star_00	AC	2674 ms	175 MB
Python	max_star_01	AC	2931 ms	161 MB
Python	small_00	AC	30 ms	12 MB
Python	small_01	AC	30 ms	12 MB
Python	small_02	AC	30 ms	12 MB
Python	small_03	AC	30 ms	12 MB
Python	small_04	AC	30 ms	12 MB
Python	sparse_random_00	AC	1310 ms	105 MB
Python	sparse_random_01	AC	1328 ms	115 MB
Python	sparse_random_02	AC	1336 ms	76 MB
Python	spfa_killer_00	AC	2410 ms	122 MB
Python	wrong_dijkstra_handmade_00	AC	30 ms	12 MB
Python	wrong_dijkstra_killer_00	AC	2199 ms	136 MB
Python	wrong_dijkstra_killer_01	AC	2509 ms	124 MB