백준 1260번 파이썬 - DFS와 BFS

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

1260번: DFS와 BFS

문제

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

코드

from collections import deque
import sys
n, m, v = map(int, sys.stdin.readline().split())
graph = [[0] * (n + 1) for _ in range(n + 1)]
visited1 = [0] * (n + 1)
visited2 = [0] * (n + 1)
for _ in range(m):
    a, b = map(int, sys.stdin.readline().split())
    graph[a][b] = graph[b][a] = 1
def dfs(s):
    visited2[s] = 1
    print(s, end=' ')
    for i in range(1, n + 1):
        if not visited2[i] and graph[s][i]:
            dfs(i)
def bfs(s):
    q = deque()
    q.append(s)
    visited1[s] = 1
    while q:
        s = q.popleft()
        print(s, end=' ')
        for i in range(1, n + 1):
            if not visited1[i] and graph[s][i]:
                q.append(i)
                visited1[i] = 1
dfs(v)
print()
bfs(v)

풀이

DFS와 BFS를 각각 사용해 방문한 노드를 출력하면 된다.

깊이 우선 탐색인 DFS와 너비 우선 탐색인 BFS를 적절히 사용하면 되고,

각각의 노트 탐색 순서의 이해를 돕는 문제같다.