기록하고 까먹지 말기

2644 본문

전공/백준

2644

yha97 2022. 11. 21. 21:31

날짜 : 2022. 11. 21

사용 언어 : python

 

문제

 

 

코드

import sys

n = int(sys.stdin.readline())  # 전체 사람의 수
visited = [False] * (n + 1)
start, end = map(int, sys.stdin.readline().split())  # 시작, 끝
cnt = 0
t = int(sys.stdin.readline())  # 관계 개수
family = [[] * (n + 1) for i in range(n + 1)]

for _ in range(t):
    a, b = map(int, sys.stdin.readline().split())
    family[a].append(b)
    family[b].append(a)


def dfs(node, move):  # 기준노드, 이동
    if node == end: # 도착
        print(move) # 출력 후 프로그램 종료
        exit(0)
    for i in family[node]:
        if not visited[i]:  # 방문하지 않았다면
            visited[i] = True
            dfs(i, move + 1)


dfs(start, 0)
print(-1) # 탈출한 경우(못찾은 경우) -1 출력

 

 

풀이

- 조건에 따라 노드를 입력받고 시작노드를 기점으로 DFS를 전개해 나간다.

- 연결된 노드끼리 탐색이 완료되었는데도 찾지 못한경우 DFS를 탈출해 -1를 출력하는 반면 도중에 찾았다면 해당 노드값을 출력한 후 exit(0)으로 해당 프로그램을 종료하는 방식으로 문제를 풀이했다.

 

 

 

알게된 점

- 앞서 풀었던 문제(11725번 https://www.acmicpc.net/problem/11725)의 복습겸 풀었던 문제다.

 

 

참고 사이트

 

'전공 > 백준' 카테고리의 다른 글

10844  (0) 2022.11.22
1502  (0) 2022.11.22
11725  (0) 2022.11.21
1735  (0) 2022.11.20
1476  (0) 2022.11.20