[programmers] 2019 카카오 개발자 겨울 인턴십 : 크레인 인형 뽑기 게임

크레인 인형 뽑기 게임

 

문제 : https://programmers.co.kr/learn/courses/30/lessons/64061

 

 

문제 설명

 

board 배열에는 행 단위로 격자안에 들어있는 각각의 인형에 해당하는 숫자가 입력된다.

board 배열의 첫 행은 격자의 가장 위쪽 행에 해당한다.

 

moves 배열의 숫자는 크레인으로 인형을 뽑는 열 번호에 해당한다.

격자에서 열 번호는 왼쪽에서 오른쪽 방향이며 가장 왼쪽이 1열에 해당한다.

 

 

 

 

코드

 

board 격자에서 인형을 꺼내 바구니에 넣은 후 해당 위치의 board 격자에는 비어있음을 의미하는 0으로 업데이트 한다.

 

def solution(board, moves):
    answer = 0
    basket = []
    for i in moves:
        if len(basket) > 1:
            if basket[-1] == basket[-2]:
                basket.pop()
                basket.pop()
                answer += 2
        for j in range(len(board)):
            if board[j][i-1] != 0:
                basket.append(board[j][i-1])
                board[j][i-1] = 0
                break

    return answer