結果

問題 No.848 なかよし旅行
ユーザー hedwig100
提出日時 2020-04-16 23:58:58
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,263 bytes
コンパイル時間 72 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 44,636 KB
最終ジャッジ日時 2024-10-03 04:25:38
合計ジャッジ時間 7,705 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 7 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 10 ** 9 + 7
INF = 10 ** 12
import sys
sys.setrecursionlimit(100000000)
dy = (-1,0,1,0)
dx = (0,1,0,-1)
from  itertools import permutations
from heapq import heapify,heappop,heappush

def dijkstra(G,s = 0): #sを始点とした最短経を求める
    n = len(G)
    dist = [INF] * n
    dist[s] = 0

    q = [(0,s)]
    heapify(q)

    while q:
        d,v = heappop(q)
        if dist[v] < d:
            continue
        for e,cost in G[v]:
            if dist[e] > dist[v] + cost:
                dist[e] = dist[v] + cost
                heappush(q,(dist[e],e))
    
    return dist

def main():
    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((b,c))
        G[b].append((a,c))
    
    d0 = dijkstra(G)
    dp = dijkstra(G,p)
    dq = dijkstra(G,q)
    if d0[p] + dp[q] + dq[0] <= t:
        print(t)
        return

    ans = -1
    for i in range(n):
        if 2 * (d0[i] + dp[i] + dq[i]) <= t:
            print(t)
            return
        if 2 * (max(dp[i],dq[i]) + d0[i]) <= t:
            ans = max(ans,t - 2 * max(dp[i],dq[i]))
    print(ans)

if __name__ =='__main__':
    main()  
0