[Python] 카카오(2) 프렌즈4블록

Updated: 2021-03-10 Categories: Programmers

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

문제설명

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

pang1

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

pang2

블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.

pang3

만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.

pang4

위 초기 배치를 문자로 표시하면 아래와 같다.

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

아이디어

주어진 board에서 삭제해야할 범위를 찾아야하기 때문에 행 단위로 한번 검사하고 열 단위로 한번 검사한다.
검사시 단어가 2개이상 연결되지 않은 부분은 빈칸으로 바꾼다.
삭제할 부분만 남게 된다면 나머지 부분의 모든 좌표를 리스트에 저장한다.
저장된 좌표값을 board에 대입하여 값을 불러오고, 그 값은 마지막 행부터 아래서 위로 순차적으로 대입한다.
위와 같은 방법을 더 이상 삭제할 범위가 없을때까지 재귀로 반복한다.

알게 된 함수/방법

copy.deepcopy(복사할 리스트) : 2차원 리스트는 기존의 .copy()함수로 복사가 되지 않는다. 따라서 2차원 리스트르 복사하려면 왼쪽의 형식에 복사하고싶은 리스트를 입력하면 복사된 리스트가 반환된다.

정답 코드

import copy
def delete_row_col(m, n, b_copy):
    for i in range(m):
        sign = False
        for j in range(n):
            if b_copy[i][j] == "_":
                continue
            if j==n-1 or b_copy[i][j] != b_copy[i][j+1]:
                if sign==True:
                    sign = False
                else:
                    b_copy[i][j]="_"                
            else:
                sign = True
    for j in range(n):
        sign = False
        for i in range(m):
            if b_copy[i][j] == "_":
                continue
            if i==m-1 or b_copy[i][j] != b_copy[i+1][j]:
                if sign==True:
                    sign = False
                else:
                    b_copy[i][j]="_"                
            else:
                sign = True
    return b_copy

def modify(m, n, board):
    b_copy = copy.deepcopy(board)  
    b_copy = delete_row_col(m, n, b_copy)
    b_copy = delete_row_col(m, n, b_copy)
    if sum(b_copy, []).count("_") == m*n:
        return sum(board, []).count("_")
    index = [[] for _ in range(n)]
    for i in range(m):
        for j in range(n):
            if b_copy[i][j] == "_":
                index[j].append((i,j))
            else:
                b_copy[i][j] = "_"
    for j in range(n):
        index[j].sort(reverse=True)
        for i in range(m-1, -1, -1):
            if not index[j]:
                break
            else:
                row = index[j][0][0]
                col = index[j][0][1]
                index[j].remove(index[j][0])
                b_copy[i][j] = board[row][col]
    return modify(m, n, b_copy)
    
def solution(m, n, board):
    for i, b in enumerate(board):
        board[i] = list(b)
    answer = modify(m, n, board)
    return answer

한재원