結果

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

ソースコード

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):
    if 2*dis_0[i]+2*max(dis_p[i],dis_q[i])<=T:
        res = max(res,T-2*max(dis_p[i],dis_q[i]))
print(res)
0