[Programmers] Python 크레인 인형뽑기 게임

https://school.programmers.co.kr/learn/courses/30/lessons/64061?language=python3 

프로그래머스

 

---

 

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

    return cnt