알고리즘
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 리턴)