[programmers] 크레인 인형 뽑기

 

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

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

 

# 주의 : 열의 모든 행을 검사 해야함

def solution(board, moves):
    answer = 0
    bucket = []
    for move in moves:
        for i in range(len(board)):
        	# moves에 저장된 열의 모든 행마다 검사
            if board[i][move-1] == 0:
                continue
            else:
                bucket.append(board[i][move-1])
                board[i][move-1] = 0 # 인형이 사라진 부분은 0으로 채우기
                break #하나의 인형만 고르면 크레인 종료
        # 같은 인형이 들어가면 삭제
        if len(bucket) >= 2 and bucket[-1] == bucket[-2]:
            bucket.pop(-1)
            answer += 1
            bucket.pop(-1)
            answer += 1
    return answer