Baekjoon #1238

Baekjoon #1238

1개의 풀이 · Python

문제 원문 보기 ↗

SOLUTION INFO

Python · main.py

main.py
from heapq import heappush, heappop
import sys
def input():
    return sys.stdin.readline().rstrip()

INF = 10000000
def dijkstra(road, start):
    dp = [INF]*(n+1)
    dp[start] = 0
    heap = []
    heappush(heap, [0, start])

    while heap:
        cur_time, cur = heappop(heap)

        if cur_time > dp[cur]: #기존 경로 사용
            continue

        for t, nxt in road[cur]:
            new_time = t + cur_time
            if new_time < dp[nxt]:
                dp[nxt] = new_time
                heappush(heap, [new_time, nxt])
    return dp

n, m, k = map(int, input().split())
road = [[] for _ in range(n+1)]
road_rev = [[] for _ in range(n+1)]
for _ in range(m):
    x, y, t = map(int, input().split())
    road[x].append([t, y])
    road_rev[y].append([t, x])
    
dp = dijkstra(road, k)
dp_rev = dijkstra(road_rev, k)

print(max([dp[i]+dp_rev[i] for i in range(1, n+1)]))

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.