18352_특정 거리의 도시 찾기.py

18352_특정 거리의 도시 찾기.py

문제 요약

N번 도시, M개의 단방향 도로
x 출발 거리가 K인 도시

풀이

deque를 이용한 풀이로 dfs와 유사한 방식으로 풀이
최단경로를 구하는 문제이기 때문에 가장 처음 방문한 경우가 최단이기 때문에 한번 방문한 적이 있으면 더이상 append를 하지 않다.

import sys
from collections import deque
input = sys.stdin.readline
n, m, k, x = map(int ,input().split())
path =[[] for _ in range(n+1)]
dist = [-1] * (n+1)
dist[x] = 0
for _ in range(m):
    start,end = map(int, input().split())
    path[start].append(end)
deq = deque()
answer = []
deq.append(x)

while deq:
    v = deq.popleft()
    for i in path[v]:
        if dist[i] == -1:
            deq.append(i)
            dist[i] = dist[v] + 1

for idx, item in enumerate(dist):
    if item == k:
        answer.append(idx)
answer.sort()
if len(answer):
    for i in answer:
        print(i)
else:
    print('-1')

풀이 2

힙큐를 이용하여 다익스트라 방식으로 통해 풀이.

import sys
import heapq
input = sys.stdin.readline
INF = int(1e9)
answer = []
n, m, k, x = map(int, input().split())
dist = [INF] * (n+1)
graph = [[] for _ in range(n+1)]
for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)

def function(s):
    q = []
    heapq.heappush(q, (0, s))
    dist[s] = 0
    while q:
        cost, city = heapq.heappop(q)
        if dist[city] < cost:
            continue
        for i in graph[city]:
            if 1 + cost < dist[i]:
                dist[i] = 1+cost
                heapq.heappush(q, (dist[i], i))
function(x)
for i, v in enumerate(dist):
    if v == k:
        answer.append(i)
if not answer:
    print('-1')
else:
    answer.sort()
    for i in answer:
        print(i)

좋은 웹페이지 즐겨찾기