[이코테] DFS/BFS - 음료수 얼려 먹기 (Python)

문제 설명

문제 조건

DFS 활용 예시 답안

n, m = map(int, input().split())

graph = []
for i in range(n):
  graph.append(list(map(int, input().split())))

# DFS로 특정 노드를 방문하고 연결된 모든 노드들도 방문
def dfs(x, y):
  # 주어진 범위를 벗어나는 경우에는 즉시 종료
  if x <= -1 or x >= n or y <= -1 or y >= m:
    return False
  
  # 현재 노드를 아직 방문하지 않았다면
  if graph[x][y] == 0:
    # 해당 노드 방문 처리
    graph[x][y] = 1
    # 상, 하, 좌, 우의 위치들도 모두 재귀적으로 호출
    dfs(x - 1, y)
    dfs(x, y - 1 )
    dfs(x + 1, y)
    dfs(x, y + 1)
    # 연결된 모든 노드들 방문한 이후에는 True값 반환
    return True
  
  # 현재 노드를 이미 방문했다면 False값 반환
  return False

result = 0
for i in range(n):
  for j in range(m):
    # 연결된 모든 노드들 방문할 때마다 카운트
    if dfs(i, j) == True:
      result += 1

print(result)

영상에 나오는 예시 답안을 정리하고 주석을 추가해봤다.
확실히 주석을 달면서 차근차근 따라가니까 이해가 좀 더 잘 되는 느낌! 🙃

BFS 방법으로도 풀 수 있다고 했는데 나중에 해보는 것으로...

문제 출처
https://www.youtube.com/watch?v=7C9RgOcvkvo&t=2706s