[프로그래머스] Lv.1 크레인 인형뽑기 게임 [Python, 파이썬] KAKAO

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/64061
• 🔑 stack

 

def solution(board, moves):
    answer = 0
    
    stack = []
    p = -1  # 스택 높이
    
    for m in moves:
        for i in range(len(board)):
            if board[i][m-1] != 0:          # 인형을 찾으면
                stack.append(board[i][m-1]) # 스택에 넣고
                board[i][m-1] = 0           # 인형을 없앤다.
                p += 1                      # 스택 높이 증가
                break
        if p >= 1:                          # 인형 2개 이상 쌓였고
            if stack[p] == stack[p-1]:      # 자신과 바로 밑 인형이 같으면
                stack.pop()                 # 둘 다 빼냄
                stack.pop()
                p -= 2                      # 높이도 조절
                answer += 2                 # 터뜨린 인형 개수 증가
    
    return answer