PRO#17679 : 프렌즈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가 사용된다.

출력 형식

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

 

 

 

---

Sol)

 

1. 프렌즈블록을 돌면서 [i][j] == [i][j+1] == [i+1][j] == [i+1][j+1] 인 부분을 기록해둔다.

2. 기록해둔 부분을  '🔥' 처리한다. (삭제를 표현)

3. 블록이 아래로 떨어져 빈 공간을 채우는 것을 표현하기 위해 '🔥'와 그 위의 값을 대체한다.

 

def solution(m, n, board):
    """
    args:
        m (int) : 높이
        n (int) : 가로
        board (list) : 판의 배치 정보
    returns:
        몇 개의 블록이 지워지는지
    """
    
    board = [list(x) for x in board]
    
    matched = True  
    while matched:
        # 삭제조건 여부 판별하고 조건 만족하는 곳은 기록
        matched = []
        for i in range(m-1):
            for j in range(n-1):
                if board[i][j] == board[i][j+1] == board[i+1][j] == board[i+1][j+1] != '🔥':
                    matched.append([i, j])
        

        # 조건 만족하는 위치 '🔥' 처리 (삭제를 표현)
        for i, j in matched:
            board[i][j] = board[i][j+1] = board[i+1][j] = board[i+1][j+1] = '🔥'
        

        # 블록이 아래로 떨어져 빈 공간을 채우는 부분 
        # 삭제한 공간('🔥')과 그 위의 값을 바꾸기로 표현
        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)

 

 

+ 석훈님의 풀이

배열을 90도 회전하는 아이디어!

def solution(m, n, board):
    board = [[j for j in i] for i in board]
    answer = 0
    switch = 1
    
    while switch:
        switch = 0
        remove_list = set()
        for i in range(m - 1):
            for j in range(n - 1):
                now_box = {
                    board[i][j],
                    board[i + 1][j + 1],
                    board[i + 1][j],
                    board[i][j + 1],
                }
                if board[i][j] != -1 and len(now_box) == 1:
                    switch = 1
                    remove_list.add((i, j))
                    remove_list.add((i + 1, j + 1))
                    remove_list.add((i, j + 1))
                    remove_list.add((i + 1, j))
        answer += len(remove_list)
        for i, j in remove_list:
            board[i][j] = -1
            
        # 배열 회전
        board = list(zip(*board))
        # -1인 블록 제거
        board = [[j for j in i if j != -1] for i in board]
        # 나머지 채우고
        board = [[-1] * (m - len(i)) + i for i in board]
        # 다시 원상복귀 회전
        board = list(map(list, zip(*board)))
        
    return answer