백준/ Silver 2 문제 , 백준 파이썬 18352 , 특정 거리의 도시 찾기[BFS]

백준/ Silver 2 문제 , 백준 파이썬 18352 , 특정 거리의 도시 찾기[BFS]

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
pypy3 로 통과

 

문제 출처

https://www.acmicpc.net/problem/18352

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

 

풀이

 

 

1. 기본 세팅

 

1. N,M,K,X 입력 받기

 

2. 도시 연결 상태를 저장할 cities 빈 리스트 선언

 

3. 양 방향이 아니므로 a->b 꼴로만 집어 넣는다.

 

 

 

 

2. BFS

 

따로 카운트를 해주기 위해 dq에 집어 넣을 때 카운트 숫자를 넣어준다.

 

이후 visited 를 사용하여 재방문을 방지하고

 

길이가 늘어나는 것을 체크하기 위해 dq 에 넣어줄때 이전 길이 + 1 로 넣어준다.

 

그러다가 원하는 길이에 해당하는 값이면 answer 리스트에 넣어주고 그 이후는 방문을 그만시키게하기 위해 dq 에 넣었던 값을 빼준다.

 

 

 

 

 

3. BFS 호출 및 처리

리턴 받은 answer 리스트가 빈값이면 -1 출력, 값이 있다면 정렬해준 뒤 값들을 출력해준다.

 

 

 

 

정답

 

 

from collections import deque

N,M,K,X=map(int,input().split())
cities=[[] for _ in range(N+1)]
for i in range(M):
    a,b=map(int,input().split())
    cities[a].append(b)

def bfs(start):
    dq=deque()
    dq.append([start,0])
    answer=[]
    visited=[False for _ in range(N+1)]
    visited[start]=True
    while dq:
        x,y=dq.popleft()
        for i in cities[x]:
            if visited[i]==False:
                dq.append([i,y+1])
                visited[i]=True
                if y+1==K:
                    answer.append(i)
                    dq.pop()
    return answer
        
result=bfs(X)

if len(result)==0:
    print(-1)
else:
    result.sort()
    for i in result:
        print(i)