[프로그래머스 lv2] [1차] 프렌즈4블록

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

출력 형식

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

입출력 예제

입출력 예

예제에 대한 설명

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

---

나의 풀이

아이디어

1. 주어진 board를 리스트로 바꾸어 블록 이동을 편하게 한다.
2. 4블록이 제거되면 블록이 내려온다. 이 과정에서의 편의를 위해 board를 시계 방향으로 회전시킨다.
3. board를 복사한 newBoard를 선언한다.
4. board를 순회하며 4블록이 발견되면 newBoard에 1로 표시한다.
   • 순회가 끝나면 newBoard에 제거될 블록들이 1로 표시되어 있다.
5. newBoard에서 1을 제거한 결과를 다시 board에 저장한다.
   • 리스트 크기 유지를 위해 빈 공간은 0으로 표시한다.
6. 더 이상 제거되는 블록이 없을 때까지 3~5를 반복한다.
7. 빈 칸을 세면 제거된 블록의 수를 알 수 있다.
   • 빈 공간을 0으로 표시했기 때문에 0을 세면 된다.

소스코드 + 주석

from copy import deepcopy

def solution(m, n, board):
	# 현재 위치, 오른쪽, 아래, 오른쪽 아래 대각선이 모두 같은 프렌즈라면 True를 반환
    def same(x, y):
        return board[x][y] != 0 and board[x][y]==board[x+1][y]==board[x][y+1]==board[x+1][y+1]
    
    # 현재 위치, 오른쪽, 아래, 오른쪽 아래 대각선에 1을 표시
    def mark(x, y):
        newBoard[x][y], newBoard[x+1][y], newBoard[x][y+1], newBoard[x+1][y+1] = 1, 1, 1, 1
    
    # 시계 방향 회전
    def rotate():
        rotated = []
        for y in range(n):
            tmp = []
            for x in range(m-1, -1, -1):
                tmp.append(board[x][y])
            rotated.append(tmp)
        return rotated
    
    # 4블록 제거
    def deleteBlocks(array):
        ret = []
        for row in array:
        	# 1이 아닌 프렌즈를 앞으로 민다
            tmp = [c for c in row if c != 1]
            
            # 제거된 1의 개수만큼 tmp 뒤에 0을 붙인다
            for _ in range(row.count(1)): tmp.append(0)
            
            # 앞으로 밀리고 빈 공간이 0으로 채워진 tmp를 ret에 넣는다
            ret.append(tmp)
        
        # 남은 프렌즈가 왼쪽으로 밀착된 결과
        return ret
    
    board = [list(row) for row in board] # 리스트로 변경
    board = rotate() # 시계 방향 회전
    m, n = n, m  # m과 n을 교환
    
    flag = True # 변화가 있으면 flag가 True가 된다. 초기값은 True
    while flag:
        flag = False # 변화가 없으면 False로 유지되어 while문을 탈출한다.
        newBoard = deepcopy(board) # 깊은 복사
        for i in range(m-1):
            for j in range(n-1):
            	# 0이 뒷쪽으로 몰렸기 때문에 0이 발견되면 더 이상 프렌즈가 없는 것
                if board[i][j+1]==0: break
                
                # 4블록이 발견되면 현재 위치에서 4블록 표시, flag에 True
                if same(i, j): mark(i, j); flag = True
        # 표시된 4블록을 제거
        board = deleteBlocks(newBoard)
    
    return sum([row.count(0) for row in board]) # board에서 0의 개수를 센다

 

소스코드만

from copy import deepcopy

def solution(m, n, board):
    def same(x, y):
        return board[x][y] != 0 and board[x][y]==board[x+1][y]==board[x][y+1]==board[x+1][y+1]
    def mark(x, y):
        newBoard[x][y], newBoard[x+1][y], newBoard[x][y+1], newBoard[x+1][y+1] = 1, 1, 1, 1
    def rotate():
        rotated = []
        for y in range(n):
            tmp = []
            for x in range(m-1, -1, -1):
                tmp.append(board[x][y])
            rotated.append(tmp)
        return rotated
    def deleteBlocks(array):
        ret = []
        for row in array:
            tmp = [c for c in row if c != 1]
            for _ in range(row.count(1): tmp.append(0)
            ret.append(tmp)
        return ret
    
    board = [list(row) for row in board] # 리스트로 변경
    board = rotate() # 시계 방향 회전
    m, n = n, m
    
    flag = True
    while flag:
        flag = False
        newBoard = deepcopy(board) # 깊은 복사
        for i in range(m-1):
            for j in range(n-1):
                if board[i][j+1]==0: break
                if same(i, j): mark(i, j); flag = True
        board = deleteBlocks(newBoard)
    
    return sum([row.count(0) for row in board])

 

테스트 결과