[백준] 1260 DFS와 BFS

문제

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

입력

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

출력

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

내 풀이

from collections import deque
import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)

n, m, start = map(int, input().split())

visited = [False]*(n+1)

graph = [[] for _ in range(n+1)]
for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)
    
for i in graph:   # sort by ascending order
    i.sort()

def DFS(start):
    visited[start] = True
    print(start, end=" ")
    
    for i in graph[start]:
        if not visited[i]:
            DFS(i)
    
def BFS(start):
    queue = deque([start])
    visited[start] = True
    while queue:
        v = queue.popleft()
        print(v, end=" ")
        for i in graph[v]:
            if not visited[i]:
                visited[i] = True
                queue.append(i)

DFS(start)
print()
visited = [False]*(n+1)   # visited 초기화
BFS(start)

코멘트

DFS와 BFS 연습하기 좋은 문제. 다만 이전에 풀었던 백준 24444 알고리즘 수업 너비 우선 탐색 1과 백준 24479 알고리즘 수업 깊이 우선 탐색 1은 무조건 시작점이 1일 때만 사용할 수 있는 것이기 때문에 이번 풀이를 DFS와 BFS 코드의 스탠다드로 기억하자.

References

https://www.acmicpc.net/problem/1260
https://ji-gwang.tistory.com/291