[프로그래머스] 프렌즈4블록 (Python)

<2018 KAKAO BLIND RECRUITMENT>

 

문제 설명

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

출력 형식

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

입출력 예제

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

예제에 대한 설명

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

---

# 4개가 모여, 지워져야 하는 좌표 리턴
def find_4(row,col,board):
    c_list = [board[row][col],board[row+1][col],board[row][col+1],board[row+1][col+1]]
    if board[row][col] != " " and min(c_list) == max(c_list):
            return [[row,col],[row+1,col],[row,col+1],[row+1,col+1]]
    return []

# 지워진 후, 빈 공간 채워서 새로운 보드 리턴
def get_new_board(row, col, board):
    while row>0:
        board[row][col],board[row-1][col] = board[row-1][col],board[row][col]
        row -= 1
    return board

def solution(m, n, board):
    answer = 0
    board = [[y for y in x] for x in board]
    while True:
        #1. 지울 부분 찾기
        erase = [] 
        for row in range(0,m-1):
            for col in range(0,n-1):
                erase_list = find_4(row,col,board)
                [erase.append(x) for x in erase_list if x not in erase]
        
        if len(erase) == 0:
            break
        else:
            #2. 지울 부분이 있다면 지우기
            for [row,col] in erase:
                board[row][col] = " "
                
            #3. 빈 공간 채운 새로운 보드 가져오기
            for row in range(m):
                for col in range(n):
                    if board[row][col] == " ":
                        board = get_new_board(row, col, board)
        answer += len(erase)          
                    
    return answer

오늘은 일찍 자야 하는데, 집안일 하다보니 시간이 너무 늦어서 풀지 않고 자려고 했는데 30분만에 쉽게 풀어서 다행이었다.

단 효율성 테스트가 있었다면 나는 큰일 났을 것 .....

나중에 다시 한번 풀어보고 싶어!

 

a,b = b,a 로 swap하는 부분과 min(char1, char2, char3, char4) 부분이 좀 마음에 들었던 풀이 !