結果

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

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline
N,M,P,Q,T = map(int, input().split())
E = [[] for _ in range(N)]
for _ in range(M):
    a,b,c = map(int, input().split())
    E[a-1].append((c, b-1))
    E[b-1].append((c, a-1))

import heapq
def dijkstra(s):
    inf = 10**18
    dist = [inf]*N
    dist[s] = 0
    h = []
    for e in E[s]:
        heapq.heappush(h, e)
    while h:
        temp = heapq.heappop(h)
        d = temp[0]
        v = temp[1]
        if dist[v] < inf:
            continue
        dist[v] = d
        for e in E[v]:
            if dist[e[1]] < inf:
                continue
            heapq.heappush(h, (e[0]+d, e[1]))
    return dist

dist_0 = dijkstra(0)
dist_p = dijkstra(P-1)
dist_q = dijkstra(Q-1)
if dist_0[P-1]+dist_p[Q-1]+dist_q[0] <= T:
    ans = T
elif dist_0[P-1]*2 > T or dist_0[Q-1]*2 > T:
    ans = -1
else:
    ans = 0
    for i in range(N):
        for j in range(N):
            if dist_0[i]+max(dist_p[i]+dist_p[j], dist_q[i]+dist_q[j])+dist_0[j] <= T:
                ans = max(ans, T-max(dist_p[i]+dist_p[j], dist_q[i]+dist_q[j]))

print(ans)
0