[Python] 카카오 2019 인턴쉽 - 인형 뽑기 게임

1. 내 풀이

def solution(board, moves):
    answer = 0
    basket = []
    for x in moves:
        x = x-1
        for y in range(len(board)) :
            if(board[y][x] == 0): continue
            else:
                if(len(basket) > 0 and basket[-1] == board[y][x]):
                    del basket[-1]
                    answer += 2
                else: basket.append(board[y][x])
                board[y][x] = 0
                break
    return answer
            

if __name__ == "__main__":
    board = [[0,0,0,0,0],[0,0,1,0,3],[0,2,5,0,1],[4,2,4,4,2],[3,5,1,3,1]]
    moves = [1,5,3,5,1,2,1,4]
    solution(board, moves)

 범위가 크지 않아서 Stack을 이욯자 않고 List를 사용하였다.