2018 KAKAO BLIND RECRUITMENT
https://school.programmers.co.kr/learn/courses/30/lessons/17679
블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈4블록".
같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.
만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.
블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.
만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.
위 초기 배치를 문자로 표시하면 아래와 같다.
TTTANT
RRFACC
RRRFCC
TRRRAA
TTMMMF
TMMTTJ
각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다
입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라.
m
, 폭 n
과 판의 배치 정보 board
가 들어온다.n
, m
≦ 30board
는 길이 n
인 문자열 m
개의 배열로 주어진다. 블록을 나타내는 문자는 대문자 A에서 Z가 사용된다.입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.
def pop_set(m, n, board):
pop_set = set()
for r in range(1, n):
for c in range(1, m):
if board[r][c] == board[r-1][c-1] == board[r-1][c] == board[r][c-1] != '_':
# pop_set.update(set([(r, c), (r-1, c-1), (r-1, c), (r, c-1)]))
pop_set |= set([(r, c), (r-1, c-1), (r-1, c), (r, c-1)])
for r, c in pop_set:
board[r][c] = 0
for idx, row in enumerate(board):
empty = ['_'] * row.count(0)
board[idx] = empty + [block for block in row if block != 0]
return len(pop_set)
def solution(m, n, board):
board = list(map(list, zip(*board)))
answer = 0
while True:
count = pop_set(m, n, board)
if count == 0: return answer
answer += count
if __name__ == '__main__':
print(solution(4, 5, ["CCBDE", "AAADE", "AAABF", "CCBBF"]))
print(solution(6, 6, ["TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"]))