[Programmers] 크레인 인형뽑기 게임

 

 

나의 코드 

 

def solution(board, moves):
    stack = []
    answer = []
    imp = []
    cnt = 0
    
    # 빈칸을 제외하여 남은 인형만 남긴 새로운 배열 만든다. 
    for i in range(len(board)):
        for t in range(len(board)):
            if board[t][i] != 0:
                stack.append(board[t][i])
        stack.reverse()
        answer.append(stack)
        stack = []
    
    # 인형을 새로운 바구니에 넣으면서 마지막 원소와 그 전 원소가 같으면 pop 시킨다. 
    # cnt 는 인형이 사라지는 횟수이기 때문에 사라진 인형의 개수는 이 값의 2배이다. 
    for l in moves:
        if len(answer[l-1]) > 0:
            imp.append(answer[l-1].pop())
            if len(imp) > 1:
                if imp[-1] == imp[-2]:
                    imp.pop()
                    imp.pop()
                    cnt += 1
        if len(answer[l-1]) == 0:
            continue
    return cnt * 2

분석 > 우선 0이 있으면 원소를 뺄 때 불편하기 때문에 각 세로줄에 남아있는 인형들로만 구성된 새로운 배열 answer을 만들었다. 그리고 이 배열을 가지고 imp 라는 새로운 바구니에 원소를 넣으면서, imp의 마지막 원소와 그 전 원소가 같은 경우 pop을 시키고 그 횟수 cnt 를 구했다. cnt 는 사라진 횟수이기 때문에 실제 사라진 인형의 개수를 구하기 위해 cnt 값에 2배를 곱한 값은 반환하여 문제를 해결하였다. 

 

 

 

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