[Programmers] 크레인 인형뽑기 게임

▶ 문제 :

 

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

 

▶ 코드 : 

from collections import deque

def solution(board, moves):
    count = 0
    st = deque()
    
    for num in moves:
        for i in range(len(board)):
            item = board[i][num-1]
            if item == 0:
                continue
            else:
                if len(st) != 0:
                    recent_item = st.popleft()
                    if recent_item == item:
                        count += 2
                    else:
                        st.appendleft(recent_item)
                        st.appendleft(item)
                else:
                    st.appendleft(item)
                board[i][num-1] = 0
                break
                
    return count

 

▶ 문제 해설 : 

1. 주어진 board에서 위쪽 행부터 탐색하며 원하는 item을 뽑아옴

2. 뽑은 item을 deque 자료구조를 이용해 저장하고 stack처럼 활용

3. deque에서 가장 최근 들어온 item을 뽑아 현재 뽑은 item과 비교하며 진행

4. 같다면 count += 2 다르다면 건너 뛰기