2021.06.18

<3> 음료수 얼려먹기(2)

답안 예시

# 얼음 틀 크기 입력
n, m = map(int, input().split())

# 얼음 틀의 형태
shape = []
for i in range(n):
  shape.append(list(map(int, input())))

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

한 시간 정도 고민했는데 결국 못풀었다.
'방문처리'라는게 익숙하지 않아서 어떤식으로 해야할지 감이 안잡혀서 막혔다..