[프로그래머스] lv.1 크레인 인형 뽑기 게임 풀이

문제

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

프로그래머스

 

문제 풀이

가장 꼭대기에 있는 인형을 뽑아서 stack에 하나씩 쌓고, 마지막 두개가 같아지면 stack에서 삭제하는 식으로 단순히 구현.

 

코드

def solution(board, moves):
    stack = []
    cnt = 0
    for move in moves:
        top_doll = None
        for i in range(len(board)): #find top_doll
            if board[i][move-1] != 0:
                top_doll = board[i][move-1]
                board[i][move-1] = 0
                break
                
        if top_doll:
            stack.append(top_doll)
            if len(stack) >= 2 and stack[-1] == stack[-2]:
                del stack[-2:]
                cnt += 2   
            
    return cnt