[프로그래머스/Python파이썬] 프렌즈4블록(2018 카카오 코딩테스트 1차)

문제 링크: programmers.co.kr/learn/courses/30/lessons/17679

코딩테스트 연습 - [1차] 프렌즈4블록

 

 

문제 설명

board 문자열 배열 입력을 리스트로 바꿔서 처리했습니다. 이 뒤에 블록이 더 이상 지워지지 않을 때까지 while문을 돌면서 블록을 지웠습니다. 

 

while문에서 한 번의 반복마다 다음을 실행합니다. 

1. mxn 크기의 remove 배열을 0으로 초기화합니다. 
2. (0, 0) 블록부터 (m-1, n-1) 블록까지 같은 모양의 2x2 블록을 찾으면 해당 인덱스의 remove 배열에 1로 표시합니다. 
3. remove에 1로 표시된 블록의 개수를 count에 저장합니다. (count : 한 텀에 지워진 블록의 개수)
4. 지워진 블록의 개수를 answer에 더합니다. 
5. 지워진 블록이 없을 경우 while문을 빠져나옵니다. 

 

 

 

소스코드

def solution(m, n, board):
    answer = 0
    for i in range(len(board)):
        board[i] = list(board[i])

    while True:
        # 같은 모양의 2X2 블록을 찾으면 remove 배열에 1로 표시
        remove = [[0]*n for _ in range(m)]
        for i in range(m - 1):
            for j in range(n - 1):
                if board[i][j] != 0 and board[i][j] == board[i][j + 1] and board[i][j] == board[i + 1][j] and board[i][j] == board[i + 1][j + 1]:
                    remove[i][j], remove[i][j + 1], remove[i + 1][j], remove[i + 1][j + 1] = 1, 1, 1, 1
        # 지워진 블록 개수 세기
        count = 0
        for i in range(m): count += sum(remove[i])
        answer += count
        # 지워진 블록이 없을 경우 break
        if count == 0: break
        # 지워진 블록을 위의 블록으로 채우기
        for i in range(m - 1, -1, -1):
            for j in range(n):
                if remove[i][j] == 1:
                    x = i - 1
                    while x >= 0 and remove[x][j] == 1: x -= 1
                    if x < 0:
                        board[i][j] = 0
                    else:
                        board[i][j] = board[x][j]
                        remove[x][j] = 1

    return answer