[백준 11559 🥇] Puyo Puyo

https://www.acmicpc.net/problem/11559

11559번: Puyo Puyo

문제

뿌요뿌요의 룰은 다음과 같다.

• 필드에 여러 가지 색깔의 뿌요를 놓는다. 뿌요는 중력의 영향을 받아 아래에 바닥이나 다른 뿌요가 나올 때까지 아래로 떨어진다.
• 뿌요를 놓고 난 후, 같은 색 뿌요가 4개 이상 상하좌우로 연결되어 있으면 연결된 같은 색 뿌요들이 한꺼번에 없어진다. 이때 1연쇄가 시작된다.
• 뿌요들이 없어지고 나서 위에 다른 뿌요들이 있다면, 역시 중력의 영향을 받아 차례대로 아래로 떨어지게 된다.
• 아래로 떨어지고 나서 다시 같은 색의 뿌요들이 4개 이상 모이게 되면 또 터지게 되는데, 터진 후 뿌요들이 내려오고 다시 터짐을 반복할 때마다 1연쇄씩 늘어난다.
• 터질 수 있는 뿌요가 여러 그룹이 있다면 동시에 터져야 하고 여러 그룹이 터지더라도 한번의 연쇄가 추가된다.

남규는 최근 뿌요뿌요 게임에 푹 빠졌다. 이 게임은 1:1로 붙는 대전게임이라 잘 쌓는 것도 중요하지만, 상대방이 터뜨린다면 연쇄가 몇 번이 될지 바로 파악할 수 있는 능력도 필요하다. 하지만 아직 실력이 부족하여 남규는 자기 필드에만 신경 쓰기 바쁘다. 상대방의 필드가 주어졌을 때, 연쇄가 몇 번 연속으로 일어날지 계산하여 남규를 도와주자!

입력

총 12개의 줄에 필드의 정보가 주어지며, 각 줄에는 6개의 문자가 있다.

이때 .은 빈공간이고 .이 아닌것은 각각의 색깔의 뿌요를 나타낸다.

R은 빨강, G는 초록, B는 파랑, P는 보라, Y는 노랑이다.

입력으로 주어지는 필드는 뿌요들이 전부 아래로 떨어진 뒤의 상태이다. 즉, 뿌요 아래에 빈 칸이 있는 경우는 없다.

출력

현재 주어진 상황에서 몇연쇄가 되는지 출력한다. 하나도 터지지 않는다면 0을 출력한다.

문제 풀이

• 주요한 알고리즘 기법은 BFS지만, 구현이나 시뮬레이션이라고 봐도 무방할 정도로 따질 것이 많은 문제였다!
• 우선, 뿌요뿌요가 위에서 아래로 떨어지기 때문에 필드 업데이트를 단순히 pop과 insert로 구현하기 위해서 예제의 필드대로 12행 6열로 저장하지 않고, 6행 12열로 저장했다. (즉, 한 열씩 행으로 저장했다.)
• while문을 돌면서, BFS를 이용해서 현재 필드에서 연쇄가 가능한 부분들을 찾는다. 
  • BFS로 탐색했을 때 4개 이상의 문자가 상하좌우로 연결된 경우라면 visited 배열을 그대로 두고, True를 반환한다.
  • 상하좌우로 연결된 문자가 4개 미만이라면 연쇄가 일어나지 않으므로, BFS 탐색 과정에서 1로 visited 값이 변경된 위치의 값들을 다시 0으로 복구하고, False를 반환한다.
• 이러한 과정을 거쳐서 현재 필드에서 연쇄가 가능한 부분들을 모두 visited 배열에 표시해놨다면, isProcessd 값이 True인 경우, 즉 BFS가 연쇄가 가능한 부분을 발견해서 True를 반환한 경우 field_update 함수를 이용해서 visited를 탐색하면서 값이 1인 경우 해당 위치의 요소를 삭제해주고, 맨 앞에 '.'을 추가하면 뿌요뿌요가 위에서 아래로 떨어지는 효과를 구현할 수 있다.
• 그리고 연쇄의 개수를 +1해준다.
• 그리고 업데이트된 필드를 기반으로 다시 for문을 돌면서 연쇄 가능한 부분을 찾고, 위와 같이 연쇄하는 과정을 반복한다.
• 더 이상 연쇄할 수 있는 부분이 없다면 for문 탐색을 끝내고도 isProcessed가 False일 것이므로, repeat를 False로 바꿔주고 while문을 종료한다.

import sys
from collections import deque

field=[[] for _ in range(6)]
# 12개의 행으로 이루어진 필드가 아니라, 열을 기준으로 6개의 행으로 필드를 만듦(field 업데이트를 pop으로 간단하게 하기 위해서)
for _ in range(12):
    line=list(sys.stdin.readline().strip())
    for idx in range(6):
        field[idx].append(line[idx])

def field_update(visited):
    global field
    for i in range(6):
        for j in range(12):
            if visited[i][j]==1:
                field[i].pop(j)
                field[i].insert(0,'.')

def BFS(i,j):
    global visited
    queue=deque()
    queue.append((i,j))
    visited[i][j]=1
    delete_list = []
    delete_list.append((i,j))

    color=field[i][j]
    directions=[(0,1),(1,0),(0,-1),(-1,0)]

    while queue:
        y,x=queue.popleft()
        for di in directions:
            dy=y+di[0]
            dx=x+di[1]
            if 0<=dy<6 and 0<=dx<12 and visited[dy][dx]==0 and field[dy][dx]==color:
                queue.append((dy,dx))
                visited[dy][dx]=1
                delete_list.append((dy,dx))
    if len(delete_list)>=4:
        return True
    else:
        # 연쇄가 일어나지 않았다면 visited 원상 복구
        for y,x in delete_list:
            visited[y][x]=0
        return False

ans=0
repeat=True
while repeat:
    isProcessed = False
    visited = [[0] * 12 for _ in range(6)]
    for i in range(6):
        for j in range(12):
            if field[i][j] != '.' and visited[i][j]==0:
                if BFS(i,j):
                    isProcessed=True
    # 모든 위치 탐색해도 BFS가 True가 안나오는 경우
    if isProcessed==False:
        repeat=False
    else:
        # 1 연쇄가 끝나면 field update
        field_update(visited)
        ans+=1

print(ans)