結果

問題 No.788 トラックの移動
ユーザー mkawa2
提出日時 2021-02-12 01:05:47
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 773 bytes
コンパイル時間 713 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 129,016 KB
最終ジャッジ日時 2024-07-18 11:44:32
合計ジャッジ時間 6,792 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other TLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

def MI(): return map(int, sys.stdin.buffer.readline().split())
def LI(): return list(map(int, sys.stdin.buffer.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
inf = 10**16

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

n, m, l = MI()
l -= 1
tt = LI()

if tt.count(0) >= n-1:
    print(0)
    exit()

e = np.array(LLI(m),dtype="i8").T
g=csr_matrix((e[2],e[:2]-1),(n,n))
# print(g)

dd=dijkstra(g,directed=False).astype("i8")

ans = inf
for g in range(n):
    mn = inf
    for s in range(n):
        if tt[s] == 0: continue
        cur = dd[l][s]-dd[s][g]
        if cur < mn: mn = cur
    mn += sum(tt[u]*dd[u][g] for u in range(n))*2
    if mn < ans: ans = mn

print(ans)
0