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

def solution(board, moves):
    answer = 0
    array = []
    for move in moves:
        i = 0
        while i < len(board):
            if board[i][move-1] != 0:
                array.append(board[i][move-1])
                board[i][move-1] = 0
                n = len(array)
                if n > 1:
                    if array[n-1] == array[n-2]:
                        for i in range(2):
                            array.pop()
                            answer += 1
                break
            else:
                i += 1

    return answer

 

게임 화면 N x N 크기의 정사각 격자

 

array = [] 를 만들어 크레인이 인형을 집었을 때 넣는 바구니를 만들었다.

len(array)가 2 이상일 때 array의 마지막 값과 마지막 바로 앞 값을 비교하여 같을 시 array.pop()을 이용하여 사라지게 한다. 인형은 두 개씩 사라지므로 answer는 두 개씩 값이 올라간다.