結果

問題 No.848 なかよし旅行
コンテスト
ユーザー hedwig100
提出日時 2020-04-17 16:16:44
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,356 bytes
コンパイル時間 127 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 51,464 KB
最終ジャッジ日時 2024-10-03 10:21:56
合計ジャッジ時間 24,803 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 4
other AC * 23 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

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
    
    for i in range(n):
        for j in range(n):
            if d0[i] + d0[j] + max(dp[i] + dp[j],dq[i] + dq[j]) <= t:
                ans = max(ans,t - max(dp[i] + dp[j],dq[i] + dq[j]))
    print(ans)

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