본문 바로가기

TIL/Algorithm

[프로그래머스] 크레인 인형뽑기 (easy, kakao)

반응형

stack을 사용하면 쉽게 풀수있는 문제

그리고 return해야할것은 사라진 인형의 개수이지, 몇번 사라지게했는지 횟수가 아님

def solution(board, moves):
    answer = 0
    stack = []
    for i in moves:
        for index, j in enumerate(board):
            if j[i-1] != 0:
                if len(stack) > 0:
                    if stack[-1] == j[i-1]:
                        answer += 2
                        stack.pop()
                    else:
                        stack.append(j[i-1])
                else:
                    stack.append(j[i-1])
                board[index][i-1] = 0
                break

    return answer

 

반응형