문제
뿌요뿌요의 룰은 다음과 같다.
필드에 여러 가지 색깔의 뿌요를 놓는다. 뿌요는 중력의 영향을 받아 아래에 바닥이나 다른 뿌요가 나올 때까지 아래로 떨어진다.
뿌요를 놓고 난 후, 같은 색 뿌요가 4개 이상 상하좌우로 연결되어 있으면 연결된 같은 색 뿌요들이 한꺼번에 없어진다. 이때 1연쇄가 시작된다.
뿌요들이 없어지고 나서 위에 다른 뿌요들이 있다면, 역시 중력의 영향을 받아 차례대로 아래로 떨어지게 된다.
아래로 떨어지고 나서 다시 같은 색의 뿌요들이 4개 이상 모이게 되면 또 터지게 되는데, 터진 후 뿌요들이 내려오고 다시 터짐을 반복할 때마다 1연쇄씩 늘어난다.
터질 수 있는 뿌요가 여러 그룹이 있다면 동시에 터져야 하고 여러 그룹이 터지더라도 한번의 연쇄가 추가된다.
남규는 최근 뿌요뿌요 게임에 푹 빠졌다. 이 게임은 1:1로 붙는 대전게임이라 잘 쌓는 것도 중요하지만, 상대방이 터뜨린다면 연쇄가 몇 번이 될지 바로 파악할 수 있는 능력도 필요하다. 하지만 아직 실력이 부족하여 남규는 자기 필드에만 신경 쓰기 바쁘다. 상대방의 필드가 주어졌을 때, 연쇄가 몇 번 연속으로 일어날지 계산하여 남규를 도와주자!
풀이
4개 이상 모여있는 색깔이 터지고 남은 자리를 위에 있는 값들로 채워야 한다.
우선 4개 이상인지 검사하는 로직은 BFS를 쓰면 수월하게 구할 수 있을 것 같아서 BFS로직을 먼저 짰고 BFS에서 큐를 순회하면서 새롭게 큐에 값을 추가할 때 인접한 곳에 같은 색깔이 있다는 것이기 때문에 cnt에 1을 더해준다.
이렇게 더해가면 연결된 같은 색상의 뿌요의 갯수를 얻을 수 있고 최종적으로 그 값을 리턴해서 cnt의 값이 4이상이라면 터져야 하므로 미리 넣어둔 tmp테이블의 row, column 값을 토대로 다 .으로 만들어준다.
.으로 만든 후에는 tmp를 answers라는 리스트에 더해준다. 여기서 만약 더해지는 값이 없다면 터지는 뿌요가 없다는 것이므로 while문을 break 해준다. break를 탈출하지 않았다면 터지는 값이 있다는 의미기 때문에 cnt를 하나 증가시켜준다.
마지막으로 answers에 담긴 값들은 터지고 자리가 빈 값이다. 이 값을 윗줄로 교체해야하기 때문에 next_row값을 현재 조회하고 있는 cur_row 보다 하나 위의 값부터 시작해서 하나씩 올려가면서 "."이 아닌 값을 찾아서 자리를 바꿔준다.
마지막으로 while문을 빠져나왔다는 건 연쇄가 끝이 났다는 것이므로 cnt를 출력해주면 된다.
from collections import deque
import sys
input = sys.stdin.readline
graph = [list(input()) for _ in range(12)]
directions = [[0, 1], [1, 0], [-1, 0], [0, -1]]
def get_same_color_cnt(color_type:str, row:int, col:int) -> int:
queue = deque()
queue.append((row, col))
visited[row][col]= 1
cnt = 1
while queue:
cur_row, cur_col = queue.popleft()
tmp.append((cur_row, cur_col))
for direction in directions:
next_row = cur_row + direction[0]
next_col = cur_col + direction[1]
if 0<= next_row < 12 and 0 <= next_col < 6 and not visited[next_row][next_col] and graph[next_row][next_col] == color_type:
visited[next_row][next_col] = 1
queue.append((next_row, next_col))
cnt+=1
return cnt
def pull_down_above_puyo() -> None:
for answer in answers:
cur_row, cur_col = answer[0], answer[1]
next_row = cur_row -1
while next_row >= 0:
if graph[next_row][cur_col] == ".":
next_row-=1
else:
graph[next_row][cur_col], graph[cur_row][cur_col] = graph[cur_row][cur_col], graph[next_row][cur_col]
next_row-=1
cur_row-=1
if __name__ == "__main__":
cnt = 0
ROW = 12
COL = 6
while True:
visited = [[0] * COL for _ in range(ROW)]
answers = []
for i in range(ROW):
for j in range(COL):
tmp = []
if not visited[i][j] and graph[i][j] != ".":
if get_same_color_cnt(graph[i][j], i, j) >= 4:
for row, col in tmp:
graph[row][col] = "."
answers+= tmp
if len(answers) == 0:
break
pull_down_above_puyo()
cnt+=1
print(cnt)