알고리즘
BFS
이경찬 :)
2023. 9. 22. 14:15
BFS - 너비 우선 탐색, 큐 자료구조에 기초한다
from collections import deque
# BFS 메서드 정의
def bfs(graph, start, visited):
# 큐(Queue) 구현을 위해 deque 라이브러리 사용
queue = deque([start])
# 현재 노드를 방문 처리
visited[start] = True
# 큐가 빌 때까지 반복
while queue:
# 큐에서 하나의 원소를 뽑아 출력
v = queue.popleft()
print(v, end = ' ')
# 해당 원소와 연결된, 아직 방문하지 않은 원소들을 큐에 삽입
for i in graph[v]:
if not visited[i]:
queue.append(i)
visited[i] = True
graph = [
[], # node 0
[2, 3, 8], # node 1
[1, 7], # node 2
[1, 4, 5], #node 3
[3, 5], # node 4
[3, 4], # node 5
[7], # node 6
[2, 6, 8], # node 7
[1, 7] # node 8
]
# 각 노드가 방문된 정보를 리스트 자료형으로 표현(1차원 리스트)
visited = [False] * 9
# 정의된 BFS 함수 호출
bfs(graph, 1, visited)
v(현재노드) | queue | 출력 |
1 | 2,3,8 | 1 |
2 | 3,8,7 | 2 |
3 | 8,7,4,5 | 3 |
8 | 7,4,5 | 8 |
7 | 4,5,6 | 7 |
4 | 5,6 | 4 |
5 | 6 | 5 |
6 | 6 |
output : 1-2-3-8-7-4-5-6
-큐 자료구조에 기초한다는 점에서 구현이 간단하다. 일반적인 경우 수행 시간은 DFS보다 좋은 편이다.