[Algorithm] (이코테) 음료수 얼려 먹기 - 파이썬

Suzie·2021년 2월 21일

BFS DFS algorithm

💭 Algorithm

목록 보기

14/49

교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 5 DFS/BFS
실전문제 5-3 음료수 얼려 먹기 149p

음료수 얼려 먹기

문제

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

입력

첫 번째 줄에 얼음 틀의 세로 길이 N과 가로 길이 M이 주어진다. (1 <= N, M <= 1,000)
두 번째 줄부터 N + 1 번째 줄까지 얼음 틀의 형태가 주어진다.
이때 구멍이 뚫려있는 부분은 0, 그렇지 않은 부분은 1이다.

출력

한 번에 만들 수 있는 아이스크림의 개수를 출력한다.

입력 예시 1

출력 예시 1

입력 예시 2

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

출력 예시2

풀이

접근 1

붙어 있는 위치 상하좌우 위치 저장하자
범위 맞춰서 맞으면 dfs 호출하는 방식
그리고 dfs 밖 메인에서 첫 자리부터 돌면서 visited 아닌데 0인 곳 찾아주기
네개 다돌았는데 dfs 못들어갔으면 pop하기

Note

ㅋㅅㅋ 일단 dfs(f.pop()[0],f.pop()[1]) 이런 짓을 해버려서 pop을 두 번이나 해서 엄청 헤맸다..ㄸㄹㄹ 진쯔...;;
tx랑 ty 계속 자리 헷갈려서., 그냥 ty, tx 이런 순으로 하면 되는데 괜히 섞어서 완전 엉켜서 시간낭비 지대로 함.. 반성하자

제출 1 - 정답

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

board = []
for i in range(n):
    tmp = list(map(str,input()))
    board.append(list(map(int, tmp)))
visited = board

f = []
cnt =0

dx = [0,1,0,-1] #북쪽부터 시계방향
dy = [-1,0,1,0]

def dfs(iy,ix):
    global visited,f
    
    visited[iy][ix] = 1
    f.append((iy,ix))

    count=0
    for i in range(len(dx)):
        tx = ix+dx[i]
        ty = iy+dy[i]
        count+=1
        if ty>=0 and ty<n and tx>=0 and tx<m:
            if visited[ty][tx] == 0:
                dfs(ty,tx)

    if count==4:
        if f:
            f.pop()
            if not f:
                return
            dfs(f[-1][0],f.pop()[1])
        else:
            return         

for i in visited:
    for j in i:
        if j==0:
            dfs(visited.index(i),i.index(j))
            cnt+=1

print(cnt)

오답노트

헤멤지수 100...ㅋㅋ 솔루션 보니까 상대적 박탈감 들어 나~~ ㅋㅋㅋㅋ
1. list(map(int,input()))이렇게 하면 다 하나씩 들어가는구나.. 몰랐다 ㅠㅠ
2. 상하좌우 따로 호출하는 거 정말.. 상상도 못해따... 뭔데 이렇게 어려운교?
뭔가 이거 이렇게 개념만 뜯어보고 담주중에 같은 방식으로 한번 짜봐야겠다... 암튼 어려워...어려워,,,어려워...

책에 있는 솔루션

# 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) # 정답 출력