結果

問題 No.1301 Strange Graph Shortest Path
ユーザー zkouzkou
提出日時 2020-11-04 22:22:21
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 958 bytes
コンパイル時間 75 ms
コンパイル使用メモリ 11,040 KB
実行使用メモリ 133,312 KB
最終ジャッジ日時 2023-10-10 21:33:45
合計ジャッジ時間 7,638 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys

import numpy as np
from scipy.sparse import csr_matrix
from scipy.sparse.csgraph import dijkstra

input = sys.stdin.buffer.readline
read = sys.stdin.buffer.read

N, M = map(int, input().split())
uvcds = list(map(int, read().split()))

us = uvcds[0::4]
vs = uvcds[1::4]
cs = uvcds[2::4]
ds = uvcds[3::4]

uv2i = {(min(u, v), max(u, v)): i for i, (u, v) in enumerate(zip(us, vs))}

costs = cs * 2
starts = us + vs
ends = vs + us

graph = csr_matrix((costs, (starts, ends)), shape=(N + 1, N + 1))
dist_vec, predecessor = dijkstra(graph, directed=False, indices=1, return_predecessors=True)

predecessor = list(predecessor)

ans = int(np.round(dist_vec[N]))
v = N
while v != 1:
    p = predecessor[v]
    if p > v:
        p, v = v, p
    i = uv2i[(p, v)]
    costs[i] = costs[i + M] = ds[i]
    v = p

# graph = csr_matrix((costs, (starts, ends)), shape=(N + 1, N + 1))
ans += int(np.round(dijkstra(graph, directed=False, indices=N)[1]))

print(ans)
0