TIL/Algorithm
2021. 1. 4.
[프로그래머스] 크레인 인형뽑기 (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