[백준] 🏅11559번 골드4 Puyo Puyo (파이썬 Python)

11559번 문제

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

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

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

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

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

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

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

 

입력

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

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

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

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

 

출력

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

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

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

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

---

문제접근

1. 같은 색 뿌요가 4개 이상 상하좌우로 연결되어 있으면 연결된 같은 색 뿌요들이 한꺼번에 없어진다.

• field 내에서 상하좌우로 연결되어있는 4개 이상의 pool을 찾아야 한다.
• BFS를 활용하면 간단하게 해결 가능하다.

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

• 없어진 뿌요들의 자리를 전부 '.'으로 대체한다.

def fall():
    for y in range(6):
        for x in range(N-2, -1, -1):
            ax = x
            if field[x][y] == '.': continue
            while ax < N-1:
                if field[ax+1][y]=='.':
                    field[ax+1][y] = field[ax][y]
                    field[ax][y] = '.'
                ax += 1

        if field[N-1][y]=='.':
            for x in range(N-1, 0, -1):
                field[x][y] = field[x-1][y]

• '.'가 아닌 뿌요가 나오면, 밑 칸을 확인한다. 해당 칸이 '.'일 경우, 해당 칸을 위 칸 값으로 채운다.
• 중력의 영향을 받아 모두 아래로 떨어져야 하므로 변수 `ax`로 마지막 행까지 '.'인지 확인하며 위 칸의 뿌요값으로 채운다.

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

• 더 이상 터질 수 있는 그룹이 없을 때까지 반복문으로 위 프로세스를 반복한다.

---

코드

# 뿌요뿌요

field = []
for _ in range(12):
    line = list(_ for _ in input())
    field.append(line)
N = len(field)
cnt = 0
dx, dy = [-1,1,0,0], [0,0,-1,1]

def find_puyo(x,y,stack,visited) -> list:
    visited[x][y] = 1
    stack.append((x,y))
    for i in range(4):
        ax = x + dx[i]
        ay = y + dy[i]
        if 0<= ax < N and 0<= ay < 6:
            if visited[ax][ay] == 1:
                continue
            elif visited[ax][ay]==0:
                visited[ax][ay] = 1
                if field[ax][ay] == field[x][y]:
                    stack = find_puyo(ax,ay,stack,visited)
    return stack

def fall():
    for y in range(6):
        for x in range(N-2, -1, -1):
            ax = x
            if field[x][y] == '.': continue
            while ax < N-1:
                if field[ax+1][y]=='.':
                    field[ax+1][y] = field[ax][y]
                    field[ax][y] = '.'
                ax += 1

        if field[N-1][y]=='.':
            for x in range(N-1, 0, -1):
                field[x][y] = field[x-1][y]

def pop_field(pop_stack):
    for xy in pop_stack:
        x, y = xy[0], xy[1]
        field[x][y] = '.' # 저장 좌표 pop
    fall()

while True:
    stack = []
    for i in range(N):
        for j in range(6):
            visited = [[0] * 6 for _ in range(N)]
            if field[i][j]=='.': continue
            pop_stack = find_puyo(i,j,[],visited)
            if len(pop_stack)>=4:
                stack.extend(pop_stack)

    if len(stack)==0:
        break
    pop_field(stack)
    cnt += 1

print(cnt)

---

느낀점

뿌요들이 없어지면 중력의 영향으로 떨어지게끔 하는 로직에서 고민을 많이 했다.

붕 떠 있는 '.'들을 모두 없애려면 `ax`를 상정해 마지막 행까지 뿌요가 떨어질 수 있도록 해야 한다.