[프로그래머스] 크레인 인형뽑기 게임 / python 파이썬

🎃 2019 카카오 개발자 겨울 인턴십 문제

 

입출력 예시

 

코드

스택으로 해결했다

def solution(board, moves):
    stack = []  # 스택
    cnt = 0  # answer
    for i in moves:  # moves를 순회하면서
        for row in range(len(board)):
            if board[row][i-1]:  # 0이 아니면, stack에 넣고 0으로 바꾸기
                stack.append(board[row][i-1])
                board[row][i-1] = 0
                break

        # 만약 stack에 같은 숫자가 있으면 삭제 후 카운트
        if len(stack) > 1:
            if stack[-1] == stack[-2]:
                cnt += 2
                stack.pop()
                stack.pop()
    return cnt