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

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

프로그래머스

---

아이디어

간단한 스택 문제이다!

bucket 리스트를 만들어 인형 번호를 넣고 인형 뽑은 자리는 0으로 만들어주고

만약 bucket에 넣을 때 인형이 맨 마지막 인형이랑 같으면 pop 해주고 answer + 1 해주면 된다

 

마지막에 answer * 2까지 해주면 완벽하다

 

---

코드

def solution(board, moves):
    answer = 0
    bucket = []
    
    for idx in moves:
        for j in range(len(board)):
            if board[j][idx-1] > 0:
                if len(bucket) >= 1 and bucket[-1] == board[j][idx-1]:
                    bucket.pop()
                    answer += 1
                else:
                    bucket.append(board[j][idx-1])
                  
                board[j][idx-1] = 0
                break
        
    return answer*2