특정 거리의 도시 찾기 (feat. 백준, 18352번)

특정 거리의 도시 찾기
https://www.acmicpc.net/problem/18352

18352번: 특정 거리의 도시 찾기

반응형

# woohyeon.kim
from collections import deque

N, M, K, X = map(int, input().split())
arr = [[] for _ in range(N + 1)]
for _ in range(M):
  f, t = map(int, input().split())
  arr[f].append(t)
node = [-1] * (N + 1)

def bfs(x):
  node[X] = 0
  q = deque()
  q.append(X)

  while q:
    fn = q.popleft()
    for tn in arr[fn]:
      if node[tn] == -1:
        node[tn] = node[fn] + 1
        q.append(tn)
  
bfs(X)

result = []
for idx in range(len(node)):
  if node[idx] == K:
    result.append(idx)
    
if len(result) == 0:
  print(-1)
else:
  for r in result:
    print(r)
# *&)*@*

시간 초과가 계속 떠서 애먹었다고 합니다....