[프로그래머스] 프렌즈4블록 /파이썬 /Python /2018 KAKAO BLIND RECRUITMENT

💡solutions )

💬 2X2 블록이 사라지고 나머지 위에 있는 블록이 떨어져 빈 공간을 채우는 로직을 보다 간단히 구현하기 위해 행열을 뒤집었다 => zip과 list로의 매핑을 통해 전치행렬 만들기

💬 four_blocks 함수에서 2X2 블록이 되는 경우의 위치를 visit_set에 담아두는데, 같은 블록이 여러 2X2에 포함될 수 있으므로 중복을 피하기 위해 set 자료형 활용

💬 모든 2x2의 경우를 찾은 후 반복문으로 1로 표시해주고 -> 행렬의 한 행마다 돌아가며 1의 개수만큼 '_'로 치환하고 맨 앞에 붙이는 것처럼 구현한다. 즉, 나머지 블록들이 서로 붙어 있게 재배치하는 것.  

 

🎫code )

def four_blocks(m, n, board):
    visit_set = set()
    for i in range(1, n):
        for j in range(1, m):
            if board[i][j] == board[i-1][j-1] == board[i-1][j] == board[i][j-1] != '_':
                    visit_set |= set([(i, j), (i-1, j-1), (i-1, j), (i, j-1)])
                    
    # 2x2가 되는 애들은 1로 표시              
    for i, j in visit_set:
        board[i][j] = 1
        
    # 1인 애들은 '_'로 대체하고 맨 앞으로 옮기는 것처럼 구현
    for idx, row in enumerate(board):
        tmp = ['_']*row.count(1)
        board[idx] = tmp + [b for b in row if b != 1]
    return len(visit_set)


def solution(m, n, board):
    answer = 0
    board = list(map(list, zip(*board))) # 행열 뒤집기
    while True:
        res = four_blocks(m, n, board)
        if res == 0:
            return answer
        answer += res

 

📌 description )

문제출처 : programmers.co.kr/learn/courses/30/lessons/17679?language=python3

코딩테스트 연습 - [1차] 프렌즈4블록

 

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

출력 형식

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