[프로그래머스/Lv.1] 크레인 인형뽑기 게임_2019 카카오 겨울 인턴십

프로그래머스

---

풀이

from collections import deque
def solution(board, moves):

    stack = deque()
    now = 0
    answer = 0

    while moves:
        col = moves.pop(0) -1

        for i in range(len(board)):
            if board[i][col]:
                now = board[i][col]
                board[i][col] = 0

                if stack and stack[-1] == now:
                    stack.pop()
                    answer += 2
                    break
                
                stack.append(now)
                break

    return answer

---

풀이과정

1. 크레인의 작동 위치에서 인형을 뽑는다.

2. 0번부터 마지막 행까지 돌면서 인형을 찾는다. : 제일 위에 있는 인형을 뽑아야하므로 0번부터 순회

3. 인형을 찾으면, 스택에 넣을지 결정한다.

• 스택에 값이 있고, 마지막 값이 현재 뽑은 인형과 같다면 → 마지막 인형을 없애고 answer에 2를 추가
• 스택에 값이 없거나, 마지막 값과 현재 뽑은 인형이 다르면 → 스택에 현재 뽑은 인형을 추가한다.