[이것이 코딩 테스트다] DFS - 음료수 얼려 먹기

YEAh·2021년 3월 9일
0
post-thumbnail

DFS
깊이 우선 탐색. 그래프에서 깊은 부분을 우선적으로 탐색하는 알고리즘
동작 원리 : 스택 / 구현 방법 : 재귀 함수 이용


✅ 문제

N, M 크기의 얼음 틀이 있다. 구멍이 뚫려 있는 부분은 0, 칸막이가 존재하는 부분은 1로 표시된다. 구멍이 뚫려 있는 부분끼리 상, 하, 좌, 우로 붙어 있는 경우 서로 연결되어 있는 것으로 간주한다. 이때 얼음 틀의 모양이 주어졌을 때 생성되는 총 아이스크림의 개수를 구하는 프로그램을 작성하시오.

입력 예시

15 14
00000111100000
11111101111110
11011101101110
11011101100000
11011111111111
11011111111100
11000000011111
01111111111111
00000000011111
01111111111000
00011111111000
00000001111000
11111111110011
11100011111111
11100011111111

출력 예시

8

➕ 문제 해설

N, M = map(int, input().split())

graph = []

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

def dfs(x, y):
    # 범위 넘어가면 False
    if x < 0 or x >= N or y < 0 or y >= M:
        return False
    # 상, 하, 좌, 우 순으로 스택에 push. 방문한 경우 False
    if graph[x][y] == 0:
        graph[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):
        # 방문하지 않은 묶음을 처음 방문했을 때 True 반환 
        # (ex. (0, 0),(0, 1) 가 0 이었다면 (0, 0)을 본다면 True를 반환하고 (0, 0)과 (0, 1)이 1로 채워진다. 다음에 (0, 1)을 본다면 이미 1로 채워졌기 때문에 False)
        if dfs(i, j) == True:
            result += 1

print(result)

📝 정리

DFS는 깊이 우선 탐색으로 재귀함수로 구현한 스택을 이용하여 문제를 해결한다.
행동할 수 있는 step을 파악하여 재귀함수로 구현하는 것이 중요했던 것 같다.

profile
End up being.

1개의 댓글

comment-user-thumbnail
2022년 9월 23일

입력값 직접 책보고 적으신 건가요? 아니면 입력값을 볼 수 있는 사이트가 있나요?

답글 달기