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

문제 정보

제목 : 프렌즈4블록

난이도 : Lv.2

사용 언어 : Python

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/17679

 

문제 설명

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

---

나의 풀이

처음 문제를 봤을 땐 어려워보이고 구현하는데 힘들 것 같았지만, 하나씩 보니 생각보다 이해하는데 어려움은 없었습니다.

m-1과 n-1을 반복하는 중첩 for문을 생성하여 board[i][j] 값을 t에 저장하고, 그것을 기준으로 오른쪽, 아래쪽, 대각선 아래쪽을 비교하여 일치하다면 좌표값을 set에 저장합니다.

set에 좌표값이 들어있다면 answer에 set의 길이만큼 저장해주고 반복문을 통해 해당 좌표에 있는 값을 []로 변경하고 set은 비워줍니다. 이후 while문 내부에 board를 순회할 반복문을 생성하여 board[i][j]에 값이 존재하고, board[i+1][j]에 빈공간([])일때 자리를 변경해줍니다. 이를 반복하다 더이상 변경할 일이 없을때 break합니다. 그리고 더이상 set에 좌표값이 저장되지 않는다면 블록이 지워지지 않는 것이므로 answer을 리턴해줍니다.

 

 

코드

def solution(m, n, board):
    answer = 0
    board = [list(x) for x in board]
    
    tmp = set()
    while True:
        
        for i in range(m-1) :
            for j in range(n-1):
                t = board[i][j]
                if ( t == [] ) :
                    continue
                else :
                    if( board[i][j+1] == t and board[i+1][j] == t and board[i+1][j+1] == t) :
                        tmp.add((i,j))
                        tmp.add((i,j+1))
                        tmp.add((i+1,j))
                        tmp.add((i+1,j+1))
                        
        if ( tmp ) :
            answer += len(tmp)
            for i,j in tmp :
                board[i][j] = []
            tmp = set()
        else :
            break
        
        while True :
            moved = 0
            for i in range(m-1) :
                for j in range(n) :
                    if( board[i][j] and board[i+1][j] == [] ) :
                        board[i][j],board[i+1][j] = board[i+1][j],board[i][j]
                        moved = 1
            if ( moved == 0 ):
                break
    
    return answer