結果

問題 No.848 なかよし旅行
ユーザー OKCH3COOHOKCH3COOH
提出日時 2019-10-30 15:09:02
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,050 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 80,716 KB
最終ジャッジ日時 2024-09-14 21:51:26
合計ジャッジ時間 6,962 ms
ジャッジサーバーID
(参考情報)
judge5 / 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 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます

ソースコード

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