크레인 인형뽑기 게임 [Level 1]

카카오 개발자 겨울 인턴십

간단히 스택을 이용하는 문제였다. 실제 있을 법한 문제와 엮어서 내서 그런지 확실히 실제 난이도와 체감난이도가 다른 듯 싶다.

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