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

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/64061

코딩테스트 연습 - 크레인 인형뽑기 게임

 

 

문제 설명

board와 moves의 정보로 크레인 작동을 시뮬레이션합니다. 이때 바구니는 stack으로 구현하여 만약 현재 뽑은 인형이 stack top의 인형과 같은 종류라면 인형을 없애야 합니다. 

 

1. doll에 현재 크레인 작동에 뽑힐 인형의 번호를 저장합니다.
2. 만약 doll == 0이면 뽑힐 인형이 없으므로 다음 크레인 작동으로 넘어갑니다. 
3. doll이 바구니에 제일 위에 있는 인형(stack top)과 같은 종류라면 doll을 바구니에 넣지않고, stack top도 pop 합니다. 
4. 그렇지 않을 경우 doll을 바구니에 push합니다. 
5. moves의 모든 원소에 대해 1~4를 반복 수행하며 사라진 인형의 개수를 셉니다. 

 

 

 

 

소스코드

def solution(board, moves):
    answer = 0
    stack = []      # 바구니

    for move in moves:
        doll = 0
        # 현재 크레인 작동에 뽑힐 인형의 번호 구하기
        for i in range(len(board)):
            if board[i][move-1] != 0:
                doll = board[i][move-1]
                board[i][move-1] = 0
                break
        if doll == 0: continue # 뽑힐 인형이 없다면 다음 크레인 작동으로 넘어감
        
        if stack and stack[-1] == doll: # 바구니에 같은 인형이 연속해서 쌓일 경우
            answer += 2
            stack.pop()
        else:
            stack.append(doll)
    
    return answer