結果

問題 No.848 なかよし旅行
ユーザー AEnAEn
提出日時 2022-12-01 11:02:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 541 ms / 2,000 ms
コード長 1,105 bytes
コンパイル時間 211 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 103,684 KB
最終ジャッジ日時 2024-10-08 11:07:46
合計ジャッジ時間 7,755 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import heappop, heappush
INF = float('inf')

def dijkstra(s, n):
    dist = [INF] * n
    hq = [(0, s)] # (distance, node)
    dist[s] = 0
    seen = [False] * n
    while hq:
        dis, v = heappop(hq)
        if dist[v] < dis:
            continue
        seen[v] = True
        for to, cost in G[v]:
            if seen[to] == False and dist[v] + cost < dist[to]:
                dist[to] = dist[v] + cost
                heappush(hq, (dist[to], to))
    return dist

N, M, P, Q, T = map(int, input().split())
G = [list() for _ in range(N)]
for i in range(M):
    a, b, c = map(int, input().split())
    a-=1;b-=1
    G[a].append((b,c))
    G[b].append((a,c))

dis_0 = dijkstra(0,N)
dis_p = dijkstra(P-1,N)
dis_q = dijkstra(Q-1,N)
if max(dis_p[0]*2,dis_q[0]*2)>T:
    exit(print(-1))

if 2*min(dis_p[0],dis_q[0])+2*dis_p[Q-1]<=T or dis_p[0]+dis_q[0]+dis_p[Q-1]<=T:
    exit(print(T))

res = 0
for i in range(N):
    for j in range(N):
        if dis_0[i]+dis_0[j]+max(dis_p[i]+dis_p[j],dis_q[i]+dis_q[j])<=T:
            res = max(res,T-max(dis_p[i]+dis_p[j],dis_q[i]+dis_q[j]))
print(res)
0