[Python] 백준 11559 Puyo Puyo (BFS)

선주·2022년 3월 11일
0

Python PS

목록 보기
58/65
post-thumbnail

📌 문제

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

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

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

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

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

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

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

입력

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

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

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

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

출력

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

예제 입력

......
......
......
......
......
......
......
......
.Y....
.YG...
RRYG..
RRYGG.

예제 출력

3

📌 풀이

💬 Code

import sys
from collections import deque
input = sys.stdin.readline


def check(x, y):
    q = deque([(x, y)])
    now = graph[x][y]
    pos = []

    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < 12 and 0 <= ny < 6 and graph[nx][ny] == now and not visited[nx][ny]:
                pos.append((nx, ny))
                q.append((nx, ny))
                visited[nx][ny] = 1

    if len(pos) >= 4:
        pos.sort(key=lambda x: (x[1], x[0]))
        for i, j in pos:
            graph[i][j] = '_'
            bombList.append((i, j))


graph = [list(input().rstrip()) for i in range(12)]
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
time = 0

while True:
    visited = [[0] * 6 for _ in range(12)]
    bombList = []

    # 색이 같은 뿌요 4개 이상 모여있는 곳 있는지 체크해서 터뜨리기
    for i in range(12):
        for j in range(6):
            if graph[i][j] != '.' and graph[i][j] != '_' and not visited[i][j]:
                check(i, j)

    if len(bombList) == 0:
        break

    # 뿌요 내리기
    for bomb in bombList:
        x, y = bomb[0], bomb[1]
        for i in range(x, 0, -1):
            graph[i][y] = graph[i-1][y]
        graph[0][y] = '.'

    time += 1

print(time)

💡 Solution

색이 같은 뿌요가 4개 이상 모여있는 곳을 찾아 뿌요를 터뜨리는데, 터뜨린 자리엔 _를 저장한다. 그리고 터뜨린 자리를 bombList에 (x, y) 형태로 삽입한다.

bombList의 길이가 0이면 터뜨린 뿌요가 없다는 것이므로 작업을 멈춘다. 하지만 0이 아니라면 터뜨린 뿌요가 있다는 것이므로 터뜨린 자리의 위에 있던 뿌요들을 내려줘야 한다.

뿌요 내리기가 관건이다. 처음에는 (11, 5)부터 (0, 0)까지 모든 좌표를 돌면서 _가 있다면 그 위의 뿌요들을 모두 내려주도록 짰었는데 시간 초과가 났다. 그래서 터진 뿌요의 좌표만 저장할 bombList를 만들고 이 리스트만 확인하면서 해당 좌표의 위에 있는 뿌요들을 한 칸씩 내려주었다.

주어진 예제의 경우를 보자.

# 뿌요 내리기
# time = 0일 때 bombList = [(10, 1), (11, 1), (10, 0), (11, 0)]
for bomb in bombList:
    x, y = bomb[0], bomb[1]
    for i in range(x, 0, -1):
        graph[i][y] = graph[i-1][y]
    graph[0][y] = '.' # 한 칸씩 내리고 나면 맨 위 칸은 당연히 비어야 함

이렇게 뿌요를 내려주는데, 내려주기 전에 먼저 bomb 정렬이 필요하다. (11, 1)을 (10, 1)보다 먼저 꺼내서 뿌요를 내리게 되면 의도했던 방향과는 전혀 다른 결과를 얻게 된다. 열이 같은 좌표끼리 모으고, 열이 같은 것 중에선 행이 작은 순서로 정렬해야 한다. (10, 1)을 (11, 1)보다 먼저 꺼내서 뿌요를 내릴 수 있도록!

→ 이걸 check 함수에서 수행한다.

def check(x, y):
    q = deque([(x, y)])
    now = graph[x][y]
    pos = []

    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < 12 and 0 <= ny < 6 and graph[nx][ny] == now and not visited[nx][ny]:
                pos.append((nx, ny))
                q.append((nx, ny))
                visited[nx][ny] = 1

    if len(pos) >= 4:
        pos.sort(key=lambda x: (x[1], x[0])) # 여기서 좌표 정렬
        for i, j in pos:
            graph[i][j] = '_'
            bombList.append((i, j))

😥 쉬워보였는데 생각보다 오래 붙들고 있었던 문제,,,
profile
기록하는 개발자 👀

0개의 댓글