[BOJ] 11559 Puyo Puyo

문제

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

필드에 여러 가지 색깔의 뿌요를 놓는다. 뿌요는 중력의 영향을 받아 아래에 바닥이나 다른 뿌요가 나올 때까지 아래로 떨어진다.

뿌요를 놓고 난 후, 같은 색 뿌요가 4개 이상 상하좌우로 연결되어 있으면 연결된 같은 색 뿌요들이 한꺼번에 없어진다. 이때 1연쇄가 시작된다.

뿌요들이 없어지고 나서 위에 다른 뿌요들이 있다면, 역시 중력의 영향을 받아 차례대로 아래로 떨어지게 된다.

아래로 떨어지고 나서 다시 같은 색의 뿌요들이 4개 이상 모이게 되면 또 터지게 되는데, 터진 후 뿌요들이 내려오고 다시 터짐을 반복할 때마다 1연쇄씩 늘어난다.

터질 수 있는 뿌요가 여러 그룹이 있다면 동시에 터져야 하고 여러 그룹이 터지더라도 한번의 연쇄가 추가된다.

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

입력

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

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

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

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

출력

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

[출처: 백준 온라인 저지]

풀이

전형적인 시뮬리이션 문제이다. 난 통상 위아래오른왼 문제가 나오면 bfs를 사용하는 것을 좋아한다. 이 문제 또한 bfs를 이용했다. 본격적인 풀이과정에 들어가기 전에 나는 필드를 가로로 보는 것이 편했기에 필드를 가로로 90도 돌린 배열인 new_puyo를 선언하여 6x12인 배열을 새로 만들었다. 그 다음 크게 두 가지 단계로 나눠서 문제를 풀었는데, 첫 번째 단계는 각 칸을 스캔하며 4방향으로 같은 색의 칸이 있는지를 bfs로 탐색하고 만약 색이 같은 칸이 있다면 boom이라는 리스트에 넣어줘서 스캔이 끝나고 boom 배열의 길이가 4이상이면 boom 배열 안의 좌표에 대하여 그 좌표에 있는 칸은 0으로 초기화를 해준다. 1단계 과정이 끝나고 나면 0을 없애줘야 하는데, 각 행에 대하여 0을 remove해주고, try문을 이용하여 에러가 뜰때까지(행에 0이 없을때까지) 계속해서 remove해줬다. 그 다음 없애준만큼 행의 앞부분을 '.'으로 채워줌으로써 중력으로 뿌요가 내려가는 기능을 구현했다.

import sys
from collections import deque
input = lambda: sys.stdin.readline().rstrip()
puyo = [list(input())for _ in range(12)]
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
new_puyo = [deque([0 for _ in range(12)])for _ in range(6)] #원래 필드를 90도로 돌린 배열
for i in range(12):
  for j in range(6):
    new_puyo[j][i] = puyo[i][j]
chain = 0
while True:
  flag = False
  for i in range(6):
    for j in range(12):
      if new_puyo[i][j] != '.' and new_puyo[i][j] != 0:
        q = deque()
        q.append((i, j, new_puyo[i][j]))
        visited = [[False for _ in range(12)]for _ in range(6)]
        visited[i][j] = True
        boom = []
        boom.append([i, j])
        while q:
          x, y, colour = q.popleft()
          for k in range(4):
            nx = x + dx[k]
            ny = y + dy[k]
            if 0 <= nx < 6 and 0 <= ny < 12:
              if not visited[nx][ny]:
                if new_puyo[nx][ny] == colour: #처음에 선택한 new_puyo[i][j]랑 같은 색이면
                  boom.append([nx, ny])
                  q.append((nx, ny, colour))
                  visited[nx][ny] = True
          if len(boom) >= 4: #같은 색인 칸이 4개 이상이면 사라져야 하기에 0으로 초기화
            flag = True
            for b in boom:
              new_puyo[b[0]][b[1]] = 0
  if not flag:
    break
  for i in range(6):
    while True:
      try:
        new_puyo[i].remove(0) #행에 0이 없어 에러가 날 때까지 remove해준다.
      except:
        break
    while len(new_puyo[i]) < 12:#없애준만큼 배열의 앞부분에 '.'을 넣어줘어야 한다
      new_puyo[i].appendleft('.')
  chain += 1
print(chain)