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	1946 ms	118 MB
Python	almost_line_01	AC	1954 ms	118 MB
Python	almost_line_02	AC	1929 ms	98 MB
Python	example_00	AC	30 ms	12 MB
Python	example_01	AC	30 ms	12 MB
Python	grid_random_00	AC	1689 ms	94 MB
Python	grid_swirl_00	AC	1954 ms	102 MB
Python	line_00	AC	2499 ms	165 MB
Python	max_dense_long_00	AC	1170 ms	78 MB
Python	max_dense_random_00	AC	1182 ms	77 MB
Python	max_dense_random_01	AC	1172 ms	77 MB
Python	max_dense_zero_00	AC	1150 ms	62 MB
Python	max_sparse_random_00	AC	1537 ms	103 MB
Python	max_sparse_random_01	AC	2043 ms	116 MB
Python	max_sparse_random_02	AC	2014 ms	116 MB
Python	max_star_00	AC	2554 ms	177 MB
Python	max_star_01	AC	2793 ms	161 MB
Python	small_00	AC	30 ms	12 MB
Python	small_01	AC	29 ms	12 MB
Python	small_02	AC	29 ms	12 MB
Python	small_03	AC	29 ms	12 MB
Python	small_04	AC	29 ms	12 MB
Python	sparse_random_00	AC	1290 ms	105 MB
Python	sparse_random_01	AC	1360 ms	115 MB
Python	sparse_random_02	AC	1315 ms	76 MB
Python	spfa_killer_00	AC	2237 ms	122 MB
Python	wrong_dijkstra_handmade_00	AC	30 ms	12 MB
Python	wrong_dijkstra_killer_00	AC	2087 ms	136 MB
Python	wrong_dijkstra_killer_01	AC	2408 ms	124 MB