結果

問題 No.848 なかよし旅行
コンテスト
ユーザー OKCH3COOH
提出日時 2019-10-30 15:09:02
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,050 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 80,716 KB
最終ジャッジ日時 2024-09-14 21:51:26
合計ジャッジ時間 6,962 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 4
other TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heappush, heappop
from scipy.sparse.csgraph import shortest_path
from scipy.sparse import lil_matrix

N, M, P, Q, T = map(int, input().split())
P -= 1
Q -= 1

edges = lil_matrix((N, N), dtype=int)

for _ in range(M):
    fr, to, cost = map(int, input().split())
    fr -= 1
    to -= 1
    edges[fr, to] = cost
    edges[to, fr] = cost

def minDist(s):
    que = [(0, s)]
    dist = [float('inf')] * N
    dist[s] = 0
    while que:
        d, now = heappop(que)
        for (_, to), cost in edges.getrow(now).todok().items():
            nd = d + cost
            if dist[to] > nd:
                dist[to] = nd
                heappush(que, (nd, to))
    return dist

minDistS = minDist(0)
minDistP = minDist(P)
minDistQ = minDist(Q)

ans = -1

if minDistS[P] + minDistP[Q] + minDistQ[0] <= T or minDistS[Q] + minDistQ[P] + minDistQ[0] <= T:
    ans = T

for mid in range(N):
    time = minDistS[mid] * 2 + max(minDistP[mid], minDistQ[mid]) * 2
    if time <= T:
        ans = max(ans, (T - time) + minDistS[mid] * 2)

print(ans)
0