結果

問題 No.788 トラックの移動
ユーザー mkawa2mkawa2
提出日時 2021-02-12 01:05:47
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 773 bytes
コンパイル時間 713 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 129,016 KB
最終ジャッジ日時 2024-07-18 11:44:32
合計ジャッジ時間 6,792 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
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 -- -
権限があれば一括ダウンロードができます

ソースコード

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