[백준 18352] 특정 거리의 도시 찾기

Junyoung Park·2022년 7월 24일
0

코딩테스트

목록 보기
508/631
post-thumbnail

1. 문제 설명

특정 거리의 문제 찾기

2. 문제 분석

전형적인 다익스트라 문제. 단방향 그래프를 그리자.

3. 나의 풀이

import sys
import heapq

n, m, k, x = map(int, sys.stdin.readline().rstrip().split())
nodes = [[] for _ in range(n+1)]
INF = sys.maxsize
for _ in range(m):
    a, b, = map(int, sys.stdin.readline().rstrip().split())
    nodes[a].append([b, 1])

def Dijkstra(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, [next_cost + cur_cost, next_node])

    return distances

distances = Dijkstra(x)
cities = [city for city, distance in enumerate(distances) if distance == k]

if not cities: print(-1)
else: print(*cities, sep='\n')

profile
JUST DO IT

0개의 댓글