結果

問題 No.848 なかよし旅行
ユーザー ttr
提出日時 2020-02-17 22:12:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 739 ms / 2,000 ms
コード長 1,042 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 103,540 KB
最終ジャッジ日時 2024-11-08 01:11:08
合計ジャッジ時間 9,880 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

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