[프로그래머스/Programmers] 크레인 인형뽑기 게임 - Python

Link

코딩테스트 연습 - 크레인 인형뽑기 게임

 

Solution

def solution(board, moves):
    answer = 0

    box = []
    for i in range(len(moves)):
        for j in range(len(board)):
            if board[j][moves[i] - 1] != 0:
                box.append(board[j][moves[i] - 1])
                board[j][moves[i] - 1] = 0

                if len(box) >= 2:
                    if box[-1] == box[-2]:
                        answer += 2
                        del box[-1]
                        del box[-1]

                break

    return answer

📌 Tip: 크레인을 작동시킨 위치에 인형이 없을 경우는 따로 작성하지 않아도 된다.