[프로그래머스] 크레인 인형뽑기 게임 문제풀이 python

mauz·2022년 6월 10일
0
post-custom-banner

🐒 문제

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

✍ 나의 풀이

def solution(board, moves):
    answer = 0
    moves = [i-1 for i in moves]
    stack = []    
    
    for move in moves:
        depth = 0

        while depth < len(board):
            if board[depth][move] == 0:
                depth += 1
            else:
                stack.append(board[depth][move])
                board[depth][move] = 0
                break

        if len(stack) > 1:
            if stack[-1] == stack[-2]:
                stack = stack[:-2]
                answer += 2
    return answer

스택 아이디어 활용

profile
쥐구멍에 볕드는 날
post-custom-banner

0개의 댓글