結果

問題 No.848 なかよし旅行
ユーザー irumo8202
提出日時 2022-02-06 22:00:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 537 ms / 2,000 ms
コード長 1,099 bytes
コンパイル時間 339 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 103,812 KB
最終ジャッジ日時 2024-06-11 14:11:13
合計ジャッジ時間 7,460 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

from heapq import heappop, heappush

INF = 1 << 60


def dijkstra(s, edge):
    n = len(edge)
    dist = [INF] * n
    dist[s] = 0
    hq = []
    heappush(hq, (dist[s], s))

    while hq:
        dist_v, v = heappop(hq)
        if dist_v > dist[v]:
            continue
        for cost, nx in edge[v]:
            if dist[nx] > dist[v] + cost:
                dist[nx] = dist[v] + cost
                heappush(hq, (dist[nx], nx))
    return dist


N, M, P, Q, T = map(int, input().split())
P -= 1
Q -= 1

G = [[] for _ in range(N)]

for _ in range(M):
    a, b, c = map(int, input().split())
    a -= 1
    b -= 1
    G[a].append((c, b))
    G[b].append((c, a))

dist_0 = dijkstra(0, G)
dist_p = dijkstra(P, G)
dist_q = dijkstra(Q, G)


tmp = dist_0[P] + dist_p[Q] + dist_q[0]
if tmp <= T:
    print(T)
    exit()

ans = -1

for u in range(N):
    for v in range(N):
        tmp = dist_0[u]
        tmp += max(dist_p[u] + dist_p[v], dist_q[u] + dist_q[v])
        tmp += dist_0[v]
        if tmp <= T:
            ans = max(ans, T - max(dist_p[u] + dist_p[v], dist_q[u] + dist_q[v]))

print(ans)
0