IT/coding study

[programmers] 크레인 인형뽑기 게임 (python)

seyeonHello 2022. 1. 10. 22:32

2019 카카오 개발자 겨울 인턴십

- 스택 사용

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