[백준/Python] 1260번: DFS와 BFS

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

1260번: DFS와 BFS

유형

그래프 탐색 (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

def dfs(v):
    visited[v] = 1
    print(v, end = ' ')
    for i in graph[v]:
        if not visited[i]:
            dfs(i)

def bfs(v):
    q = deque()
    q.append(v)
    visited[v] = 1
    while q:
        x = q.popleft()
        print(x, end = ' ')
        for i in graph[x]:
            if not visited[i]:
                q.append(i)
                visited[i] = 1

n, m, v = map(int, input().split())
graph = [[] for _ in range(n + 1)]
visited = [0] * (n + 1)
for _ in range(m):
    x, y = map(int, input().split())
    graph[x].append(y)
    graph[y].append(x)

for i in range(1, n + 1):
    graph[i].sort()

dfs(v)
print()
visited = [0] * (n + 1)
bfs(v)

문제풀이

1. queue를 사용하기위해 collections 모듈을 사용한다. 

2. dfs(깊이우선탐색) 함수를 생성한다.

2-1) 방문했음을 visited 리스트에 표시한다.

2-2) 인자 v를 출력한다.

2-3) for 반복문을 사용하여 v에 인접한 엣지들을 확인한다. 방문한 적이 없다면 dfs함수의 인자로하여 재귀함수로 더 깊이 탐색한다. 

3. bfs(너비우선탐색) 함수를 생성한다. 

3-1) queue를 생성하고 함수의 인자를 append한다.

3-2) 방문했음을 visited 리스트에 표시한다. 

3-3) queue가 빌 때까지 while 반복문을 사용한다.

3-4) q의 가장 왼쪽 값을 제거 및 출력하고 해당 값의 인접한 엣지들을 확인한다. 방문한 적이 없다면 queue에 append한다.

4. 입력값을 사용해 graph를 생성하고 visited 리스트를 0으로 초기화한다.

5. 번호가 낮은 것부터 방문해야 하므로 graph를 정렬한다.

6. dfs함수를 호출하고 개행한다.

7. visited를 초기화하고 bfs를 호출한다.