結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー zkou
提出日時 2020-11-04 22:22:21
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 958 bytes
コンパイル時間 417 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 107,332 KB
最終ジャッジ日時 2024-09-13 00:35:33
合計ジャッジ時間 8,540 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other TLE * 1 -- * 32
権限があれば一括ダウンロードができます

ソースコード

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