[Python] 백준 1260번: 첫 DFS와 BFS 구현

📎 백준 1260번 풀러 가기!

1260번: DFS와 BFS

문제

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

입력

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

출력

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

---

오늘은 DFS와 BFS이다.

 

나는 그래프 문제를 단 한 번도! 풀어본 적이 없다.그래서 그래프 문제에 매번 자신이 없었는데,계속 이렇게 회피할 수는 없지 싶어가장 기본인 DFS와 BFS 문제를 풀어보았다.

 

자료구조 시간이나 알고리즘 시간에항상 DFS와 BFS 개념에 대해서만 짚고 넘어가고,실제로 구현을 해본 것은 이번이 처음이다.

 

이론도 중요하지만, 실제로 구현하는 것은 또 느낌이 다른 것 같다.머릿속으로 그려지는 DFS와 BFS의 작동 방식을어떻게 코드로 담아내느냐는 다른 문제라고 해야하나

 

아무튼 이번에 구현을 해보았으니앞으로 더 많이 구현해보면서 점차 익숙해져나가야겠다.

---

아래는 내가 구현한 코드👇

import sys
from collections import deque

N, M, V = map(int, sys.stdin.readline().split())
graph = [[] for _ in range(N + 1)]
visited = [False] * (N + 1)

for _ in range(M):
    start, end = map(int, sys.stdin.readline().split())
    graph[start].append(end)
    graph[end].append(start)

for x in range(1, N + 1):
    graph[x].sort()

def dfs(n):
    print(n, end = ' ')
    visited[n] = True
    for x in graph[n]:
        if not visited[x]:
            dfs(x)

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

dfs(V)
print()

visited = [False] * (N + 1)
bfs(V)