문제

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 프렌즈4블록. 같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.

friends4block

만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.

friends4block2

블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.
만약 빈 공간을 채운 후에 다시 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가 사용된다.
출력 형식
입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.

접근법

편한 연산을 위해 각 문자열들을 모두 리스트로 변환후 먼저 없어져야할 좌표들을 찾았다.
중복은 set()을 통해 없앴고 해당 좌표에는 0을 집어넣었다.
문제는 블록들이 없어지고 난 후 위의 블록들이 떨어져야 하는것을 구현하는 것이었는데 계산을 쉽게 하기위해 가로와 세로를 바꾸어 계산 후 다시 가로-세로 반전을 해주었다. 이것을 구현한 함수가 drop()이다.
더 간단하게 구현할 수 있었을 것 같은데 너무 빙 돌아간 느낌이다. 흠..

코드

def solution(m, n, board):
    cnt = 0
    board2 = []
    for i in board:
        board2.append(list(i))
    while 1:
        remove_block = []
        for i in range(1, m):
            for j in range(1, n):
                if board2[i-1][j-1] == board2[i][j-1] == board2[i-1][j] == board2[i][j] and board2[i][j] != 0:
                    remove_block.append((i-1,j-1)) 
                    remove_block.append((i,j-1))
                    remove_block.append((i-1,j))
                    remove_block.append((i,j))
        if remove_block == []:
            break
        cnt += len(set(remove_block))
        for i in set(remove_block):
            board2[i[0]][i[1]] = 0
        # print(board2)
        # print(drop(board2))
        board2 = drop(board2)
    return cnt

def drop(matrix):
    idx = 0
    dropped = []
    while idx < len(matrix[0]): #세로 가로 변환
        tmp = []
        for i in matrix:
            tmp.append(i[idx])
        dropped.append(tmp)
        idx += 1
    for i in range(len(dropped)):
        cnt_0 = 0
        remained_block = []
        for j in dropped[i]:
            if j == 0:
                cnt_0 += 1
            else:
                remained_block.append(j)
        dropped[i] = [0]*cnt_0 + remained_block
    idx2 = 0
    dropped2 = []
    while idx2 < len(dropped[0]): #변환
        tmp2 = []
        for i in dropped:
            tmp2.append(i[idx2])
        dropped2.append(tmp2)
        idx2 += 1
    return dropped2

다른사람풀이

def solution(m, n, board):
    x = board
    x2 =[]

    for i in x: 
        x1 = []
        for i2 in i:
            x1.append(i2)
        x2.append(x1)

    point = 1
    while point != 0:
        list = []
        point = 0
        for i in range(m - 1):
            for j in range(n - 1):
                if x2[i][j] == x2[i][j + 1] == x2[i + 1][j] == x2[i + 1][j + 1] != '팡!':
                    list.append([i, j])
                    point += 1

        for i2 in list:
            i, j = i2[0], i2[1]
            x2[i][j], x2[i][j + 1], x2[i + 1][j], x2[i + 1][j + 1] = '팡!', '팡!', '팡!', '팡!'

        for i3 in range(m):
            for i in range(m - 1):
                for j in range(n):
                    if x2[i + 1][j] == '팡!':
                        x2[i + 1][j], x2[i][j] = x2[i][j], '팡!'

    cnt = 0
    for i in x2:
        cnt += i.count('팡!')
    return cnt