[백준 1238] 파티

Junyoung Park·2022년 3월 11일
0

코딩테스트

목록 보기
242/631
post-thumbnail

1. 문제 설명

파티

2. 문제 분석

특정 노드 i -> 도착 노드 x, 도착 노드 x -> 특정 노드 i까지 가는 경로의 합이 최댓값인 노드 i를 구해야 한다. 양방향 간선이 아니기 때문에 각 i마다 x에 대한 최단 거리를 구하기 위해 다익스트라 알고리즘을 사용해야 한다.

3. 나의 풀이

import sys
import heapq

INF = sys.maxsize

n, m, x = map(int, sys.stdin.readline().rstrip().split())
nodes = [[] for _ in range(n+1)]
for _ in range(m):
    tail, head, time = map(int, sys.stdin.readline().rstrip().split())
    nodes[tail].append([head, time])

def Dijsktra(start):
    distances = [INF for _ in range(n+1)]
    distances[start] = 0

    pq = []
    heapq.heappush(pq, [0, start])

    while pq:
        cur_cost, cur_node = heapq.heappop(pq)

        if distances[cur_node] < cur_cost: continue

        for next_node, next_cost in nodes[cur_node]:
            if distances[next_node] > cur_cost+next_cost:
                distances[next_node] = cur_cost+next_cost
                heapq.heappush(pq, [cur_cost+next_cost, next_node])

    return distances

distances_x = Dijsktra(x)

for i in range(1, n+1):
    if i == x: continue
    distances_i = Dijsktra(i)
    distances_x[i] += distances_i[x]

print(max(distances_x[1:]))
profile
JUST DO IT

0개의 댓글