문제 6 - 프렌즈 블록

출처 : 카카오 신입 공채 1차 코딩 테스트 문제 해설 – tech.kakao.com

카카오 신입 공채 1차 코딩 테스트 문제 해설

 

---

 

6. 프렌즈 4블록(난이도: 상)

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 “프렌즈 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가 사용된다.

출력 형식

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

 

---

[ 문제 설명 ]

 

이 문제는 차근차근 3단계를 단계로 구현 문제를 풀이 풀이하기가 쉽습니다.

 

1) 일치 여부를 찾는 것

 

2) 일치한 위치를 삭제

 

3) 빈 공간 위에 있는 블록을 아래로 떨어 뜨리게 하기

 

주석과 함께 코드를 읽어 내려가시면 문제를 이해하실 수 있을 듯합니다.

 

---

 

[ 코드 ]

 

# 프렌즈 블록
from typing import List


def solution(m: int, n: int, board: List[str]) -> int:
    board = [list(x) for x in board]

    matched = True
    while matched:
        # 1) 일치 여부 판별
        matched = []
        for i in range(m - 1):  # 맨 밑단(m줄) 까지 왔다면 어차피 2x2 매칭이 안될테니까 m-1 까지
            for j in range(n - 1):  # 마찬가지 원리
                if board[i][j] == \
                        board[i][j + 1] == \
                        board[i + 1][j + 1] == \
                        board[i + 1][j] != '#':  # "#" 은 빈공간을 의미 하므로
                    matched.append([i, j])

        # 2) 일치한 위치 삭제
        for i, j in matched:
            board[i][j] = board[i][j + 1] = board[i + 1][j + 1] = board[i + 1][j] = '#'  # 2x2이 매칭이 되면 빈 공간으로 바꿔주기

        # 3) 빈공간 블럭 처리
        for _ in range(m):
            for i in range(m - 1):
                for j in range(n):
                    if board[i + 1][j] == '#':
                        board[i + 1][j], board[i][j] = board[i][j], '#'  # 빈 공간 위에 블록이 있다면 위치 바꾸기

    return sum(x.count('#') for x in board)


m, n = 4, 5
board = ["CCBDE", "AAADE", "AAABF", "CCBBF"]
sol = solution(m, n, board)
print(sol)

 

---

후기

 

그럭저럭 풀만 합니다.