n개의 \b도시에 단방향이며 거리가 1인 m개의 도로가 있다. x 도시부터 출발하여 최단거리가 k인 도시들을 출력하는 문제. 힙과 INF를 이용하는 다익스트라 방법을 이용하여 풀었으며 입력되는 도로들을 인접리스트 형식으로 사용하였다. x로부터 모든 도시들의 이동 거리를 배열에 저장 후 filter을 사용하여 거리가 k인 도시들을 저장 후 출력하여 풀었다.
\nimport heapq\nimport sys\ninput = sys.stdin.readline\nINF = sys.maxsize\n\ndef dijkstra():\n dist = [INF for _ in range(n+1)]\n dist[x] = 0\n\n q = []\n heapq.heappush(q, [0, x])\n while q:\n cnt, now = heapq.heappop(q)\n for nn, nd in graph[now]:\n nextd = nd + cnt\n if dist[nn] > nextd:\n heapq.heappush(q, [nextd, nn])\n dist[nn] = nextd\n return dist\n\nn, m, k, x = map(int, input().split())\ngraph = {i: [] for i in range(1, n+1)}\n\nfor _ in range(m):\n s, e = map(int, input().split())\n graph[s].append([e, 1])\n\ndist = dijkstra()\nans = list(filter(lambda i: dist[i] == k, range(1, n+1)))\n[print(i) for i in ans] if len(ans) else print(-1)