Python - DFS (깊이 우선 탐색)

- DFS

Depth-First Search의 약자로 그래프에서 깊은 부분을 우선적으로 탐색하는 알고리즘

 

• 스택 혹은 재귀 함수를 이용하며, 구체적인 동작은 다음과 같다.

1) 탐색 시작 노드를 스택에 삽입하고 방문 처리를 한다.

 

2) 스택의 최상단 노드에 방문하지 않은 인접한 노드가 하나라도 있으면

그 노드를 스택에 넣고 방문처리한다. 방문하지 않은 인접 노드가 없으면 스택에서 최상단 노드를 꺼낸다.

 

3) 더 이상 2번의 과정을 수행할 수 없을 때까지 반복한다.

 

def dfs(graph, v, visited):  # (그래프 리스트, 현재 노드, 노드의 방문여부 리스트)
    visited[v] = True		# 처음 노드인 v의 방문을 True 처리
    print(v, end=' ')
    
    for i in graph[v]:
    	if not visited[i]		# 노드가 방문 리스트에 없으면
        	dfs(graph, i, visited)	# 재귀적으로 방문 처리
            
graph = [       # 각 노드가 연결된 정보를 표현 (그래프를 2차원 리스트로 표현)
    [],
    [2, 3, 8],
    [1, 7],
    [1, 4, 5],
    [3, 5],
    [3, 4],
    [7],
    [2, 6, 8],
    [1, 7]
]

visited = [False] * 9 # 1~8을 인덱스로 사용하기 위해 9로 초기화, 각 노드가 방문된 정보 표현
dfs(graph, 1, visited)

 

 

참고: https://www.youtube.com/watch?v=7C9RgOcvkvo