結果

問題 No.788 トラックの移動
ユーザー SSlime
提出日時 2020-06-17 15:54:05
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,002 bytes
コンパイル時間 86 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 25,536 KB
最終ジャッジ日時 2024-07-03 12:15:14
合計ジャッジ時間 6,790 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other TLE * 1 -- * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,m,l = map(int,input().split())
l -= 1
T = list(map(int,input().split()))
from heapq import *
def Dijkstra(n,path,start):
    inf = float('inf')
    d = [inf]*n
    d[start] = 0
    pq = [(0, start)]
    heapify(pq)
    while pq:
        dist, p = heappop(pq)
        if dist > d[p]:continue
        for nxt,cost in path[p].items():
            if dist + cost >= d[nxt]:continue
            d[nxt] = dist + cost
            heappush(pq,(d[nxt],nxt))
    return d
path = [{} for _ in range(n)]
for i in range(m):
    a,b,c = map(int,input().split())
    a -= 1
    b -= 1
    path[a][b] = c
    path[b][a] = c

ld = Dijkstra(n,path,l)
def total(n,path,t,start,ld):
    d = Dijkstra(n,path,start)
    res = 0
    delta = float('inf')
    for i in range(n):
        res += 2*d[i]*t[i]
        if t[i] > 0:
            delta = min(delta, ld[i] - d[i])
    if res == 0:return res
    else:
        return res + delta
ans = float('inf')
for i in range(n):
    ans = min(ans,total(n,path,T,i,ld))
print(ans)
0