[알고리즘] 프로그래머스 크레인 인형뽑기 게임 / python

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

프로그래머스

 

 

def solution(board, moves):
    answer = 0
    ret = []
    for move in moves:
        for i in range(len(board)):
            if board[i][move-1]:
                if ret and ret[-1] == board[i][move-1]:
                    ret.pop()
                    board[i][move-1] = 0
                    answer += 2
                    break
                else:
                    ret.append(board[i][move-1])
                    board[i][move-1] = 0
                    break
    return answer