結果

問題 No.848 なかよし旅行
ユーザー OKCH3COOH
提出日時 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 4
other RE * 26
権限があれば一括ダウンロードができます

ソースコード

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