[프로그래머스/Level 2] [1차] 프렌즈4블록 - python

{1차]프렌즈 4블록

문제내용 블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈4블록". 같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다. 각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다 입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라. 입력방식 - 입력으로는 str1과 str2의 두 문자열이 들어온다. 각 문자열의 길이는 2 이상, 1,000 이하이다. - 2 ≦ n, m ≦ 30 - board는 길이 n인 문자열 m개의 배열로 주어진다. 블록을 나타내는 문자는 대문자 A에서 Z가 사용된다.

---

프로그래밍 순서

- ... 추후에...

---

전체코딩

- pytest 모듈을 이용해서 코드가 원하는 정답이 나오는지 바로 확인 가능

- pytest 사용법

pip install -U pytest
pytest --version # 설치 확인
 
pytest [python 파일 이름] -s # 올바르게 코딩했다면 PASSED 출력

- 코드

def solution(m, n, board):
    board=[list(ele) for ele in board]
    while True:
        remove_b = find_puzzle(m, n, board)
        if remove_b == []: break
        board = remove_block(board, remove_b)
        board = change_loc(m, n, board)
    return count_answer(board)
 
def find_puzzle(m, n, board):
    remove_b = []
    m_index = 0 # 행
    n_index = 0 # 열
    
    while m_index != m:
        check = board[m_index][n_index]
        try: 
            if check != 0 and check == board[m_index][n_index+1] == check == board[m_index+1][n_index] == check == board[m_index+1][n_index+1]:
                remove_b += [(m_index, n_index), (m_index, n_index+1), (m_index+1, n_index), (m_index+1, n_index+1)]
        except IndexError:
            pass
            
        if n_index + 1 == n:
            m_index += 1
            n_index = 0
        else:
            n_index += 1
            
    return remove_b
 
def remove_block(board, remove_list):
    for ele in remove_list:
        board[ele[0]][ele[1]] = 0
    return board
 
def change_loc(m, n, board):
    m_index = m-1 # 행
    n_index = n-1 # 열
    while m_index != -1:
        if board[m_index][n_index] == 0:
            for num in range(m_index, -1, -1):
                if board[num][n_index] != 0:
                    board[m_index][n_index] = board[num][n_index]
                    board[num][n_index] = 0
                    break
        if n_index - 1 == -1:
            m_index -= 1
            n_index = n-1
        else:
            n_index -= 1
    return board
 
def count_answer(board):
    num = 0 
    for ele in board:
        num += ele.count(0)
    return num
 
def test():
    assert solution(4, 5, ["CCBDE", "AAADE", "AAABF", "CCBBF"]) == 14
    assert solution(6, 6,     ["TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"]) == 15