알고리즘/프로그래머스
[프로그래머스] 크레인 인형뽑기 /파이썬
빙빙
2021. 6. 27. 11:15
def solution(board, moves):
answer = 0
bag = []
for k in moves:#moves원소안에 있는걸로
i = k-1 #세로
for j in range(len(board)): #가로
# print(i,j)
if board[j][i] == 0: #0일 땐 넘김
continue
else: #0이 아닐때
bag.append(board[j][i])
board[j][i] = 0 #0으로 대체해주기
if len(bag) >= 2:
b1 = bag.pop(-1) #맨끝
b2 = bag.pop(-1) #두번째 끝
if b1 == b2: #앞꺼와 같으면 없애고 카운트
answer += 2
else:
bag.append(b2) #아니면 두번째 끝꺼부터 다시 넣어줌
bag.append(b1)
break #바구니에 넣었으면 다음 moves로 넘김
return answer