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

문제

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

프로그래머스

 

풀이

스택을 사용해 바구니의 맨 마지막에 크레인이 집은 인형과 같은 인형이 담겨 있다면 pop()을 해줍니다.

 

코드

파이썬

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