[프로그래머스-Lv1] 크레인 인형뽑기 / Python

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

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

import sys

input = sys.stdin.readline
n = int(input())
board = [list(map(int, input().split())) for _ in range(n)]
moves = list(map(int, input().split()))


def solution(board, moves):
    basket = []
    answer = 0
    for i in range(len(moves)):
        for j in range(len(board)):
            if board[j][moves[i] - 1] == 0:
                continue
            basket.append(board[j][moves[i] - 1])
            board[j][moves[i] - 1] = 0
            if len(basket) != 1 and basket[len(basket) - 1] == basket[len(basket) - 2]:
                del basket[len(basket) - 1]
                del basket[len(basket) - 1]
                answer += 2
            break
    return answer


print(solution(board, moves))

 

해설

크레인의 인덱스에 인형이 위치하면 해당 인형 위치의 값을 0으로 바꾸고 그 값을 basket에 넣어준다.

이때 basket에 마지막 값과 바로 이전 값이 같으면 해당 데이터를 지워주고 answer를 2씩 늘려준다.