DFS, BFS
2021. 6. 22. 22:11ㆍalgorithm
반응형
DFS
- 스택
- 재귀함수 이용
# DFS 메서드 정의
def dfs(graph, v, visited):
# 현재 노드를 방문 처리
visited[v] = True
print(v, end = '')
# 현재 노드와 연결된 다른 노드를 재귀적으로 방문
for i in graph[v]:
if not visited[i]:
dfs(graph, i, visited)
BFS
- 큐
from collections import deque
# BFS 메서드 정의
def bfs(graph, start, visited):
queue = deque([start])
# 현재 노드를 방문 처리
visited[start] = True
# 큐가 빌 때까지 반복
while queue:
# 큐에서 하나의 원소를 뽑아 출력
v = deque.popleft()
print(v, end = ' ')
# 해당 원소와 연결된, 아직 방문하지 않은 원소들을 큐에 삽입
for i in graph[v]:
if not visited[i]:
queue.append(i)
visited[i] = True
출처: 이것이 취업을 위한 코딩테스트다 with 파이썬
반응형
'algorithm' 카테고리의 다른 글
백준 바이러스 코드 및 해설 (파이썬) (0) | 2021.06.22 |
---|---|
프로그래머스 네트워크 코드 및 해설 (파이썬) (1) | 2021.06.22 |
백준 트리 순회 코드 및 해설 (파이썬) (0) | 2021.06.08 |
백준 회전하는 큐 코드 및 해설 (파이썬) (0) | 2021.06.08 |
프로그래머스 로또의 최고 순위와 최저 순위 코드 및 해설 (파이썬) (0) | 2021.06.08 |