백준 11559번 Puyo Puyo | python

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

11559번: Puyo Puyo

 

문제

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

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

 

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

입력

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

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

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

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

출력

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

 

---

이번 문제는 BFS를 이용하는 문제이다.

2차원 배열 loop를 돌면서 ‘.’가 아닌 값이 경우, BFS를 수행하여 해당 row, col의 색과 같은 것의 갯수를 측정하고 이게 4개 이상인 경우에 제거하는 방식을 진행한다.

그리고 4개 이상 제거한 경우에 flag를 True로 바꿔주고 제거된 자리에 블록들을 내려주는 작업을 진행해줍니다.  문제에서 다수의 그룹이 터져도 다같이 터지는 경우에도 한 번으로 카운팅 한다고 했으므로 따로 flag가 True일 때만 answer + 1을 해줍니다.

이런식으로 진행을 하고 flag가 False인 경우에는 while loop를 나오는 방식으로 진행합니다.

아래의 코드를 참고

 

import sys
from collections import deque

input = sys.stdin.readline

boards = [list(input().rstrip()) for _ in range(12)]
answer = 0


def BFS(row, col, boards):
    queue = deque()
    queue.append((row, col))
    visited = [[False for _ in range(6)] for _ in range(12)]
    visited[row][col] = True
    tmp = [(row, col)]
    while queue:
        r, c = queue.popleft()
        for j, i in [[1, 0], [0, 1], [-1, 0], [0, -1]]:
            nj, ni = r + j, c + i
            if 0 <= nj < 12 and 0 <= ni < 6 and not visited[nj][ni] and boards[row][col] == boards[nj][ni]:
                tmp.append((nj, ni))
                visited[nj][ni] = True
                queue.append((nj, ni))

    # 길이가 4이상이면 블록을 .으로 바꿔주고 True
    if len(tmp) >= 4:
        for y, x in tmp:
            boards[y][x] = '.'
        return True
    return False


while True:
    flag = False    # 4개 이상의 그룹이 제거 유무 판단하는 boolean
    # loop를 돌면서 '.'이 아닌 위치에서 BFS 진행
    for row in range(11, -1, -1):
        for col in range(6):
            if boards[row][col] != '.':
                if BFS(row, col, boards):
                    flag = True

    #   flag가 True이면 중력을 통해 블록들을 내려준다. False이면 while loop 탈출
    if flag:
        answer += 1
        for col in range(6):
            cnt, row = 0, 11
            while row >= 0:
                if boards[row][col] == '.':
                    cnt += 1
                else:
                    boards[row + cnt][col], boards[row][col] = boards[row][col], boards[row + cnt][col]
                    row += cnt
                    cnt = 0
                row -= 1
    else:
        break

print(answer)

 

커피 한 잔은 더 좋은 게시글을 쓰는 데 도움이 됩니다.