[카카오 기출문제] '프렌즈4블록' 문제 풀이 - Python

Contents

1. 문제 설명
   
   [제한사항]
   
   [입출력 예]
2. 알고리즘 분석 
   
   [나의 풀이]

---

 

문제 설명

 

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 프렌즈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가 들어온다.
• 2 ≦ n, m ≦ 30
• board는 길이 n인 문자열 m개의 배열로 주어진다. 블록을 나타내는 문자는 대문자 A에서 Z가 사용된다.

 

입출력 예

m	n	board	answer
4	5	["CCBDE", "AAADE", "AAABF", "CCBBF"]	14
6	6	["TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"]	15

알고리즘 분석

 

• 나의 풀이

import copy
D_BLOCK = "@"  # 빈 공간을 표시하는 문자 

# 보드를 회전시킨다.
def turn_board(m, n, board):
    new_board = [[0] * m for i in range(n)]
    for i in range(m):
        for j in range(n):
            new_board[j][i] = board[i][j]
    return new_board

# 블록을 이동시킨다.
def move_block(m, n, board):
    new_board = turn_board(m, n, board)  # 보드를 변경하기 쉽게 돌려줌 
    for i in range(n):
    	# 해당 행을 문자열로 변환 후 D_BLOCK을 공백으로 변환. 마지막으로 부족한 공간은 D_BLOCK으로 채움
        new_board[i] = ''.join(new_board[i]).replace(D_BLOCK, "").rjust(m, D_BLOCK)
    board = turn_board(n, m, new_board)  # 보드를 원래대로 돌림 
    return board

# 조건에 맞는 블록을 지운다.
def delete_block(m, n, board):
    new_board = copy.deepcopy(board)
    for i in range(m-1):
        for j in range(n-1):
            block = board[i][j]
            # board의 4개의 블록이 같다면 new_board의 4개의 블록을 D_BLOCK으로 대체 
            if block == board[i+1][j] and block == board[i][j+1] and block == board[i+1][j+1]:
                new_board[i][j] = new_board[i+1][j] = new_board[i][j+1] = new_board[i+1][j+1] = D_BLOCK
    return new_board

# D_BLOCK의 개수를 센다. 
def count_block(m, board):
    answer = 0
    for i in range(m):
        answer += board[i].count(D_BLOCK)  # 해당 행에 있는 D_BLOCK의 개수를 answer에 더함
    return answer

def solution(m, n, board):
    board = list(map(list, board))
    while True:
        new_board = delete_block(m, n, board)  # 조건에 맞는 블록을 제거
        if board == new_board:  # 새로운 보드와 기존의 보드가 같다면
            break  # 반복문 탈출 
        board = move_block(m, n, new_board)  # 빈 공간이 생긴 보드의 블록을 이동
    answer = count_block(m, board)  # 빈 공간의 개수를 카운트 
    print(answer)
    return answer