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

문제 링크

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

구현 문제로 문제 내용 그대로 코드로 구현했기에 자세한 설명은 생략한다.

def solution(board, moves):
    answer = 0
    n = len(board[0])
    stack = []
    
    for pick in moves:
        i = 0
        while i < n:
            if board[i][pick - 1] != 0:
                stack.append(board[i][pick - 1])
                board[i][pick - 1] = 0
                
                if len(stack) >= 2 and stack[-1] == stack[-2]:
                    stack.pop()
                    stack.pop()
                    answer = answer + 2
                break
                
            i = i + 1
                    
    return answer