[프로그래머스 Lv.2] 프렌즈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가 사용된다.

출력 형식

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

입출력 예제

예제에 대한 설명

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

---

풀이

단순 구현으로 풀었다.

1. 모든 요소 한 번 씩 순회한다. (마지막 열과 행의 데이터는 검사하지 않는다. -> 0 ~ m-1, 0 ~ n-1)

2. 만약 요소의 오른쪽 데이터와, 아래 데이터, 대각선 아래 데이터의 값이 같으면 set에 넣는다.

3. set에 있는 블록을 없애고, 블록을 재배치한다.

4. 위의 과정을 더 이상 사라지는 블록이 없을 때 까지 반복한다. (무한반복, set에 추가된 내용이 없으면 종료)

 

처음 아이디어는 위와 같았는데 풀면서 한 가지가 추가됐다.

2번에 데이터가 0이 아닐 경우를 추가해주었다. (0은 사라지고 없는 값이다.)

임의로 정했다ㅎㅎ

그럼 아이디어를 구체적으로 살펴보자.

 

아이디어 1 : set 자료형을 사용하자!

set 자료형의 중복을 허용하지 않는 특성을 이용하였다.

위와 같은 상황에서 겹쳐진 데이터를 중복해서 제거하지 않기 위해 사용하였다. 

                if (
                    board[i][j] == board[i][j + 1]
                    and board[i][j] == board[i + 1][j]
                    and board[i][j] == board[i + 1][j + 1]
                    and board[i][j] != 0
                ):
                    pos.add((i, j))
                    pos.add((i, j + 1))
                    pos.add((i + 1, j))
                    pos.add((i + 1, j + 1))

위처럼 특별한 검사 없이 마음놓고 add 하면 된다.

 

아이디어 2 : pop_block!

블록을 이동시키기 위해 아래와 같은 메서드를 작성했다.

def pop_block(board, i, j):
    board[i][j] = 0
    for ii in range(i, 0, -1):
        board[ii][j] = board[ii - 1][j]
    board[0][j] = 0
    return board

좌표를 주면 해당 위치를 없애고 차례로 이동시키는 코드이다.

이 코드를 작성하고 테스트 케이스를 돌려보니 오류가 떴다. 이유는 set에 순서가 없기 때문이었다.

그래서 정렬을 해 주었다.

p = sorted(pos, key=lambda x: x[0])

위에서부터 제대로 이동하게 되었다!

 

전체 코드

def pop_block(board, i, j):
    board[i][j] = 0
    for ii in range(i, 0, -1):
        board[ii][j] = board[ii - 1][j]
    board[0][j] = 0
    return board


def solution(m, n, board):
    answer = 0
    board = [list(x) for x in board]
    while True:
        pos = set()
        for i in range(m - 1):
            for j in range(n - 1):
                if (
                    board[i][j] == board[i][j + 1]
                    and board[i][j] == board[i + 1][j]
                    and board[i][j] == board[i + 1][j + 1]
                    and board[i][j] != 0
                ):
                    pos.add((i, j))
                    pos.add((i, j + 1))
                    pos.add((i + 1, j))
                    pos.add((i + 1, j + 1))

        if pos:
            p = sorted(pos, key=lambda x: x[0])
            for i, j in p:
                board = pop_block(board, i, j)
                answer += 1
        else:
            break
    return answer


print(solution(4, 5, ["CCBDE", "AAADE", "AAABF", "CCBBF"]))