結果

問題 No.1301 Strange Graph Shortest Path
コンテスト
ユーザー zkou
提出日時 2020-11-04 22:06:25
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 901 bytes
コンパイル時間 384 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 104,020 KB
最終ジャッジ日時 2024-09-13 00:34:03
合計ジャッジ時間 12,387 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 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))}

graph = csr_matrix((cs + cs, (us + vs, vs + us)), shape=(N + 1, N + 1), dtype=np.int64)
dist_vec, predecessor = dijkstra(graph, directed=False, indices=1, return_predecessors=True)

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)]
    cs[i] = ds[i]
    v = p

graph = csr_matrix((cs + cs, (us + vs, vs + us)), shape=(N + 1, N + 1), dtype=np.int64)
ans += int(np.round(dijkstra(graph, directed=False, indices=N)[1]))

print(ans)
0