[프로그래머스] 코딩테스트 연습2019 카카오 개발자 겨울 인턴십 크레인 인형뽑기 게임

문제

문제 설명

게임개발자인 "죠르디"는 크레인 인형뽑기 기계를 모바일 게임으로 만들려고 합니다.
"죠르디"는 게임의 재미를 높이기 위해 화면 구성과 규칙을 다음과 같이 게임 로직에 반영하려고 합니다.

게임 화면은 "1 x 1" 크기의 칸들로 이루어진 "N x N" 크기의 정사각 격자이며 위쪽에는 크레인이 있고 오른쪽에는 바구니가 있습니다. (위 그림은 "5 x 5" 크기의 예시입니다). 각 격자 칸에는 다양한 인형이 들어 있으며 인형이 없는 칸은 빈칸입니다. 모든 인형은 "1 x 1" 크기의 격자 한 칸을 차지하며 격자의 가장 아래 칸부터 차곡차곡 쌓여 있습니다. 게임 사용자는 크레인을 좌우로 움직여서 멈춘 위치에서 가장 위에 있는 인형을 집어 올릴 수 있습니다. 집어 올린 인형은 바구니에 쌓이게 되는 데, 이때 바구니의 가장 아래 칸부터 인형이 순서대로 쌓이게 됩니다. 다음 그림은 [1번, 5번, 3번] 위치에서 순서대로 인형을 집어 올려 바구니에 담은 모습입니다.

만약 같은 모양의 인형 두 개가 바구니에 연속해서 쌓이게 되면 두 인형은 터뜨려지면서 바구니에서 사라지게 됩니다. 위 상태에서 이어서 [5번] 위치에서 인형을 집어 바구니에 쌓으면 같은 모양 인형 두 개가 없어집니다.

크레인 작동 시 인형이 집어지지 않는 경우는 없으나 만약 인형이 없는 곳에서 크레인을 작동시키는 경우에는 아무런 일도 일어나지 않습니다. 또한 바구니는 모든 인형이 들어갈 수 있을 만큼 충분히 크다고 가정합니다. (그림에서는 화면표시 제약으로 5칸만으로 표현하였음)

게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.

 

입력 & 출력

입출력 예

boardmovesresult

[[0,0,0,0,0],[0,0,1,0,3],[0,2,5,0,1],[4,2,4,4,2],[3,5,1,3,1]]

[1,5,3,5,1,2,1,4]

4

풀이 과정

1. 크레인의 구조와 바구니의 모습을 보자마자 스택을 떠올리게 되었다. 하지만 인형을빼는 크레인에서는 스택형식으로 값을빼버리면 배열이 흐트러져 버리는 모습을 볼 수 있기 때문에 이것은 크레인이 집어 올리는 위치의 값을 0으로 바꾸는 형식으로 기획을 변경하였다.

2. 일단 실제 크레인이 움직인다고 생각하고 구성하여 보았을때 moves는 배열로 이루어져 있었지만 각각의 순서에 맞는 위치만 생각하면 되기에 moves를 for문을 이용해서 한개씩 위치값을 얻는 것을 시작으로 생각한다.

3. moves를 한개씩 가져온 후 어떤 위치의 숫자에 접근하여야 하는지를 선택이 된다면 그 위치의 값을 result라는 리스트에 저장을 하고 동시에 값을 0으로 변경하여 준다.

4. 이곳에 조건문을 이용해서 결과 값의 길이가 2 이상일때 이전 숫자와 값을 비교하여 같을 경우에는 현재 숫자와 이전 숫자 모두를 pop하는 방식으로 배열을 관리하고 이때 빠져나가는 숫자는 한쌍씩 빠져나가게 되므로 2씩을 cnt 변수에 입력해준다.

5. 마지막으로 cnt를 리턴하여 주면 원하는 결과를 얻을 수 있다. 

 

코드

def solution(board, moves):
    result = []
    cnt = 0
    for move in moves:
        for i, bo in enumerate(board):
            if not bo[move - 1] == 0:
                result.append(board[i][move - 1])
                board[i][move - 1] = 0
                if len(result) > 1:
                    if result[-2] == result[-1]:
                        result.pop(-2)
                        result.pop(-1)
                        cnt += 2
                break

    return cnt