알고리즘

DFS) 음료수 얼려 먹기

이경찬 :) 2023. 9. 22. 14:35
# n, m을 공백으로 구분하여 입력받기
n, m = map(int, input().split())

# 2차원 리스트의 맵 정보 이어받기
graph = []

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

# 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)
    return True
  return False

# 모든 노드에 대하여 음료수 채우기
result = 0
for i in range(n):
  for j in range(m):
    # 현재 위치에서 DFS 수행
    if dfs(i,j) == True:
      result += 1
print(result)

1. 얼음판의 사이즈를 n, m으로 받는다

2. 0,1로 얼음판을 입력받는다 (한줄씩)

3. DFS수행 (0으로 연결된 모든 노드를 방문처리 후 최종 적으로 True 리턴)