TIL/Algorithm
[프로그래머스] 크레인 인형뽑기 (easy, kakao)
재융
2021. 1. 4. 19:55
반응형
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
반응형