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	1952 ms	118 MB
Python	almost_line_01	AC	1945 ms	118 MB
Python	almost_line_02	AC	1884 ms	98 MB
Python	example_00	AC	29 ms	12 MB
Python	example_01	AC	29 ms	12 MB
Python	grid_random_00	AC	1667 ms	94 MB
Python	grid_swirl_00	AC	1974 ms	102 MB
Python	line_00	AC	2441 ms	165 MB
Python	max_dense_long_00	AC	1161 ms	78 MB
Python	max_dense_random_00	AC	1154 ms	77 MB
Python	max_dense_random_01	AC	1167 ms	77 MB
Python	max_dense_zero_00	AC	1113 ms	61 MB
Python	max_sparse_random_00	AC	1466 ms	103 MB
Python	max_sparse_random_01	AC	1951 ms	116 MB
Python	max_sparse_random_02	AC	1921 ms	116 MB
Python	max_star_00	AC	2545 ms	176 MB
Python	max_star_01	AC	2810 ms	161 MB
Python	small_00	AC	29 ms	12 MB
Python	small_01	AC	29 ms	12 MB
Python	small_02	AC	29 ms	12 MB
Python	small_03	AC	28 ms	12 MB
Python	small_04	AC	29 ms	12 MB
Python	sparse_random_00	AC	1282 ms	105 MB
Python	sparse_random_01	AC	1330 ms	115 MB
Python	sparse_random_02	AC	1306 ms	76 MB
Python	spfa_killer_00	AC	2293 ms	121 MB
Python	wrong_dijkstra_handmade_00	AC	29 ms	12 MB
Python	wrong_dijkstra_killer_00	AC	2159 ms	136 MB
Python	wrong_dijkstra_killer_01	AC	2487 ms	124 MB