[BOJ] 1260번: DFS와 BFS

■ 문제 풀이

dfs와 bfs 알고리즘의 정석 풀이

■ 문제 해답

from collections import deque
import sys
input = sys.stdin.readline

# n = 정점의 개수 / m = 간선의 개수 / v = 시작할 정점
n, m, v = map(int, input().split())

graph = [[] for i in range(n + 1)]
for i in range(m):
  a, b = map(int, input().split())
  graph[a].append(b)
  graph[b].append(a)
visited_dfs = [False] * (n + 1)
visited_bfs = [False] * (n + 1)

def dfs(v):
  visited_dfs[v] = True
  print(v, end = ' ')

  for i in graph[v]:
    if not visited_dfs[i]:
      dfs(i)

def bfs(v):
  q = deque([v])
  visited_bfs[v] = True
  while q:
    node = q.popleft()
    print(node, end = " ")
    for i in graph[node]:
      if not visited_bfs[i]:
        q.append(i)
        visited_bfs[i] = True

dfs(v)
print()
bfs(v)

문제출처: https://www.acmicpc.net/problem/1260

1260번: DFS와 BFS