[백준 2176] 합리적인 이동경로

Junyoung Park·2022년 3월 22일
0

코딩테스트

목록 보기
310/631
post-thumbnail

1. 문제 설명

합리적인 이동경로

2. 문제 분석

다익스트라를 통해 T 기준 최단 거리를 구한다. 이를 통해 특정 노드에서 T를 가는 경로보다 다른 경로(즉 특정 노드와 연결된 다른 노드 사용한 경로) 거리가 더 짧다면 이를 합리적인 이동경로라고 한다. DP를 통해 시작 노드 1번의 합리적인 이동경로를 구한다.

3. 나의 풀이

import sys
import heapq

n, m = map(int, sys.stdin.readline().rstrip().split())
nodes = [[] for _ in range(n+1)]
for _ in range(m):
    a, b, c = map(int, sys.stdin.readline().rstrip().split())
    nodes[a].append([b, c])
    nodes[b].append([a, c])
INF = sys.maxsize
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 = Dijsktra(2)
# T 기준 다익스트라 알고리즘

def rational_path(cur_node):
    if dp[cur_node] == 0:
        for next_node, next_cost in nodes[cur_node]:
            if distances[cur_node] > distances[next_node]:
                # cur_node보다 next_node 사용이 T에 도달하는 더 합리적인 이동 경로.
                dp[cur_node] += rational_path(next_node)
        return dp[cur_node]
    else:
        return dp[cur_node]


dp = [0 for _ in range(n+1)]
dp[2] = 1
# T -> T일 때 1로 취급
print(rational_path(1))
profile
JUST DO IT

0개의 댓글