프로그래머스 (Level 1) : 크레인 인형뽑기/ Python / 2019 카카오 개발자 겨울 인턴십 코딩테스트

문제

코딩테스트 연습 - 크레인 인형뽑기 게임

 

출처 : 프로그래머스 - 크레인 인형뽑기 문제

 

그림이 있어서 일단 풀기 싫었지만 레벨 1이기도 하고 문제를 읽어보니 쉬워보여서 풀어보았다.

그림처럼 인형 정보가 있는 2차원 배열에서 movies 배열로 명령하는 열의 인형을 맨 위에것 부터 하나씩 뽑으면 된다.

큐를 이용해 풀었다.

 

 

풀이

from collections import deque

def solution(board, moves):
    
    answer = 0  
    dq=deque()
    for m in moves:
        y = m-1

        for x in range(len(board)):
            if board[x][y]!=0:
                tmp=board[x][y]
                board[x][y]=0
                
                if dq and dq[-1]==tmp:
                    answer+=2
                    dq.pop()               
                else:
                    dq.append(tmp)
                
                break #한번만 수행
    return answer