[프로그래머스] 크레인 인형뽑기

cheeeese·2022년 1월 16일
0

코딩테스트 연습

목록 보기
27/151
post-thumbnail

📖 문제

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

💻 내 코드

def solution(board, moves):
    
    answer = 0
    myStack = []
    
    for move in moves:
        for i in range(len(board)):
            if board[i][move-1] != 0:
                myStack.append(board[i][move-1])
                board[i][move-1] = 0
    
                if len(myStack) > 1:
                    if myStack[-1]==myStack[-2]:
                        myStack.pop(-1)
                        myStack.pop(-1)
                        answer+=2
                break
            
    return answer

➕ 추가

  • 처음 실수했던 부분
if myStack[-1]==myStack[-2]:
	myStack.pop(-1)
   	myStack.pop(-1)

여기서

	myStack.pop(-1)
   	myStack.pop(-2)

라고 썼었는데 하나 pop하면 그 뒤에 있던 index는 앞당겨지기 때문에 -1, -1로 해줘야함..........

0개의 댓글