프로그래머스 [level2] 프렌치4블록- 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가 사용된다.

출력 형식

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

입출력 예제

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개의 블록이 지워진다.

문제 코드

def solution(m, n, board):
    answer = 0
    check = []
    new_board = []

    for i in board:
        a = list(i.strip())
        new_board.append(a)  # 보드 배열화
    while True:
        check = []
        for i in range(m - 1):  # y
            for j in range(n - 1):  # x
                if new_board[i][j] == "0":  # 탐색한 좌표가 0일시 패스
                    continue

                if new_board[i][j] == new_board[i][j + 1]:
                    if new_board[i][j] == new_board[i + 1][j] and new_board[i][j + 1] == new_board[i + 1][
                        j + 1]:  # x,y좌표 2*2 탐색했을 때 전부 같은 값일 때
                        check.append((i, j))
                        check.append((i, j + 1))
                        check.append((i + 1, j))
                        check.append((i + 1, j + 1))
        if len(check) == 0:
            break
        else:
            answer += len(set(check))

            for c in check:
                new_board[c[0]][c[1]] = "0"  # 터질 블록들 0으로 초기화

            for c in reversed(check):  # 블록들 내리기 (밑에 있는 좌표부터 탐색할수 있도록 reversed로 뒤집음)
                check_n = c[0] - 1  # 터진 좌표로부터 하나하나씩 값을 바꿀 좌표
                put_n = c[0]  # 터진좌표를 저장할 값

                while check_n >= 0:  # 터진 자리 위에 있는 블록들을 다 내린다.
                    if new_board[put_n][c[1]] == "0" and new_board[check_n][c[1]] != "0":
                        new_board[put_n][c[1]] = new_board[check_n][c[1]]
                        new_board[check_n][c[1]] = "0"
                        put_n -= 1

                    check_n -= 1

    return answer

문제 해석

먼저 22의 배열을 탐색하며 터질블록을 찾아야 한다. 해당좌표에서 22로 탐색했을 때 조건에 맞으면 터질 좌표는 표시를 한후 count를 해준다. 그후 전체 배열의 탐색이 끝나면 표시되어있는 블록들을 터트려준다. 그후 터진 좌표에서 y축기준으로 위로 올라가며 터지지 않은 블록들을 탐색을 해야하니까 -1을 해주면서 아직 터지지 않은 블록을 터진 좌표로 바꿔준다. 그후 다시 while문을 돌려 모든 블록들이 다 터지지 않을때까지 반복문으로 돌린다.