크레인 인형뽑기 게임

크레인 인형뽑기 게임

stack = []

def solution(board, moves):
    answer = 0
    l = len(board)
    
    for move in moves:
        for i in range(l):  # i = 0 ~ 4
            if board[i][move-1]:
                doll = board[i][move-1]
                board[i][move-1] = 0
                
                if not stack or stack[-1] != doll:
                    stack.append(doll)
                elif stack and stack[-1] == doll:
                    answer += 2
                    stack.pop()
                break
                
    return answer

2023.07.14

def solution(board, moves):
    answer = 0
    s = list()
    
    for m in moves:
        for i in range(len(board)):
            if board[i][m-1] != 0:
                if s and s[-1] == board[i][m-1]:
                    s.pop()
                    answer += 2
                else:
                    s.append(board[i][m-1])
                board[i][m-1] = 0
                break
    return answer