CodingTest/Programers

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

JunJangE 2022. 8. 29. 12:51

문제

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

알고리즘

- 반복문을 통해 크레인의 위치를 이동한다.

- 반복문을 통해 크레인으로 뽑을 수 있는 인형을 확인 후 뽑는다.

- 뽑은 인형의 개수가 2개 이상이라면 맨 위에 2개의 인형을 비교한다.

- 비교 후 같다면 pop() 한 후 인형의 개수만큼 카운트한다.

코드

def solution(board, moves):
    answer = 0
    pick_up = []
    # 반복문을 통해 크레인의 위치를 이동
    for move in moves:
        # 크레인이 뽑을 수 있는 인형을 확인
        for i in range(len(board)):
            if board[i][move-1] != 0:
                pick_up.append(board[i][move-1])
                board[i][move-1] = 0
                
                if len(pick_up) >= 2:
                    if pick_up[-1] == pick_up[-2]:
                        pick_up.pop()
                        pick_up.pop()
                        answer += 2
                break
                
    return

github