DFS, BFS

2021. 6. 22. 22:11algorithm

반응형

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 파이썬

반응형