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

list의 행과 열을 바꾸기 위해 numpy 를 이용해서 transpose 해주고 stack을 이용해서 답 구하기 !

 

Question

 

게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.

 

 

Code

 

import numpy as np
def solution(board, moves):
    answer = 0
    stack = []
    board = np.array(board).T
    board = (board).tolist()
    board = [[i for i in row if i!=0] for row in board] # 0(인형없음) 제거.
    
    for m in moves:
        if board[m-1] : 
            stack.append(board[m-1].pop(0)) # stack에 인형 넣고 board[m-1]에서 해당 인형 제거.
        
        if len(stack)>=2 and stack[-1]==stack[-2] :
            answer += 2
            stack.pop()
            stack.pop()
    
    return answer

 

now me

On my github