結果

問題 No.848 なかよし旅行
ユーザー OKCH3COOHOKCH3COOH
提出日時 2019-10-30 15:18:48
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,098 bytes
コンパイル時間 367 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 67,952 KB
最終ジャッジ日時 2024-09-14 21:53:01
合計ジャッジ時間 3,536 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
権限があれば一括ダウンロードができます

ソースコード

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 = [[] for _ in range(N)]

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

def minDist(s):
    dist = [float('inf')] * N
    dist[s] = 0
    que = [(0, s)]
    while que:
        d, now = heappop(que)
        for to, cost in edges[now]:
            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] + minDistP[0] <= T:
    ans = T

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

print(ans)
0