프로그래머스 - 크레인 인형뽑기 게임

문제

 

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

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

 

 

 

풀이

스택 자료구조를 이용하여 문제를 해결한다.

0이 아닌 숫자들은 스택에 저장하고 스택에 저장되어 있는 숫자들 중에 상위 2개가 같은지 비교한다.

 

 

 

코드

python

def solution(board, moves):
    answer = 0
    board = list(map(list, zip(*board)))
    board = [[i for i in item if i != 0] for item in board]
    s = []
 
    for i in moves:
        if len(board[i - 1]) == 0:
            continue
        s.append(board[i - 1].pop(0))
        if s[-1] == 0:
            s.pop(-1)
        while len(s) >= 2 and s[-1] == s[-2]:
            s.pop(-1)
            s.pop(-1)
            answer += 2
 
    return answer