2021.07.18 18일차 문제 예시답안

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

• board 배열은 2차원 배열로 크기는 "5 x 5" 이상 "30 x 30" 이하입니다.
• board의 각 칸에는 0 이상 100 이하인 정수가 담겨있습니다.
  • 0은 빈 칸을 나타냅니다.
  • 1 ~ 100의 각 숫자는 각기 다른 인형의 모양을 의미하며 같은 숫자는 같은 모양의 인형을 나타냅니다.
• moves 배열의 크기는 1 이상 1,000 이하입니다.
• moves 배열 각 원소들의 값은 1 이상이며 board 배열의 가로 크기 이하인 자연수입니다.

def solution(board, moves):
    answer = []
    answer = 0

    for i in moves:
        for j in range(len(board)):
            if board[j][i-1] != 0: #j가 먼저 와야함, i는 1보다 큼
                a.append(board[j][i-1])
                board[j][i-1] = 0
                break

        if len(a) >= 2 and a[-1] == a[-2]: #out of index, len 2보다 김
            a.pop()
            a.pop()
            answer += 2

    return answer