블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈4블록".
같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.
만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.
블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.
만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.
위 초기 배치를 문자로 표시하면 아래와 같다.
각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다
입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라.
from copy import deepcopy
def remove(new_board):
# copy
tmp_board = deepcopy(new_board)
# row, col length
m = len(tmp_board)
n = len(tmp_board[0])
for i in range(m-1):
for j in range(n-1):
block = new_board[i][j]
check_same = all(map(lambda x: block == x, (new_board[i][j+1], new_board[i+1][j], new_board[i+1][j+1])))
if check_same:
for r, c in ([i,j], [i+1, j], [i, j+1], [i+1, j+1]):
tmp_board[r][c] = ''
return tmp_board
def drop(tmp_board):
tmp_board = list(zip(*tmp_board))
for i in range(len(tmp_board)):
tb = tmp_board[i]
remove_count = tb.count('')
tmp_board[i] = [''] * remove_count + list(filter(lambda x: x!='', tb))
tmp_board = list(map(list, zip(*tmp_board)))
return tmp_board
def solution(m, n, board):
new_board = []
for b in board:
new_board.append(list(b))
tmp_board = remove(new_board)
tmp_board = drop(tmp_board)
while new_board != tmp_board:
new_board = tmp_board
tmp_board = remove(new_board)
tmp_board = drop(tmp_board)
return sum(new_board, []).count('')