[2018 카카오] 프렌즈4블록

  • 2018 카카오 블라인드 공채 코딩테스트 프렌즈4블록 문제 풀이입니다.
  • 본 문제는 프로그래머스 홈페이지에서 직접 풀어볼 수 있습니다.

프렌즈4블록 문제 설명

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 “프렌즈4블록”. 같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.

Problem Fig 1

만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.

Problem Fig 2

블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.

Problem Fig 3

만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.

Problem Fig 4

위 초기 배치를 문자로 표시하면 아래와 같다.

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가 사용된다.

출력 형식

입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.

입출력 예제

mnboardanswer
45[“CCBDE”, “AAADE”, “AAABF”, “CCBBF”]14
66[“TTTANT”, “RRFACC”, “RRRFCC”, “TRRRAA”, “TTMMMF”, “TMMTTJ”]15

예제에 대한 설명

  • 입출력 예제 1의 경우, 첫 번째에는 A 블록 6개가 지워지고, 두 번째에는 B 블록 4개와 C 블록 4개가 지워져, 모두 14개의 블록이 지워진다.
  • 입출력 예제 2는 본문 설명에 있는 그림을 옮긴 것이다. 11개와 4개의 블록이 차례로 지워지며, 모두 15개의 블록이 지워진다.

프렌즈4블록 문제 풀이

  • Python set()을 활용하면 쉽게 문제를 풀 수 있습니다.
    • 문제에 중복된 영역이 존재할 가능성이 있음을 보여줬으므로, 중복을 허용하지 않는 set()을 사용합니다.
  • 4개의 블록이 동일한지 판단하기 위해 인덱스를 사용하면 편합니다.
    • 그러기 위해서는 1D List를 2D List로 변환해야 합니다. (그래야 i, j를 동시에 사용할 수 있기 때문입니다.)
    • 또한 리스트의 첫 번째 위치를 기준으로 오른쪽, 하단, 하단 오른쪽 블록을 비교합니다. 이러한 방법은 인덱스의 범위를 초과할 수 있기 때문에 주어지는 mn의 크기를 1 줄입니다.
  • 모든 게임은 제거 가능한 블록이 없을 때까지 반복하도록 합니다.
def solution(m, n, board):
    answer = 0

    # get 2d list
    board = list(map(list, zip(*board)))

    def number_of_deleted(m, n, board):
        del_set = set()

        # print(board)

        # search 2x2 blocks
        for i in range(0, n - 1):
            for j in range(0, m - 1):
                if board[i][j] == board[i + 1][j] == board[i][j + 1] == board[i + 1][j + 1] != '*':
                    del_set.update(
                        [(i, j), (i + 1, j), (i, j + 1), (i + 1, j + 1)])

        # board
        for i, j in del_set:
            board[i][j] = 0
        for i, row in enumerate(board):
            empty = ['*'] * row.count(0)
            board[i] = empty + [block for block in row if block != 0]
        return len(del_set)

    while True:
        pop = number_of_deleted(m, n, board)
        if pop == 0:
            return answer
        answer += pop