[프로그래머스] 크레인 인형뽑기 게임 / Python / 2019 KAKAO 겨울 인턴십

https://programmers.co.kr/learn/courses/30/lessons/64061?language=python3

프로그래머스

---

---

Solution

	def game(line, board, basket):
	cnt = 0
	for i in range(len(board)):
	if board[i][line] == 0:
	continue
	else:
	if len(basket) == 0:
	basket.append(board[i][line])
	else:
	if basket[-1] == board[i][line]:
	basket.pop(-1)
	cnt += 2
	else:
	basket.append(board[i][line])
	board[i][line] = 0
	return cnt
	return cnt
	def solution(board, moves):
	n = len(board)
	basket = []
	answer = 0
	for i in moves:
	answer += game(i-1, board, basket)
	return answer

view raw 크레인인형뽑기게임.py hosted with ❤ by GitHub

문제 유형 : 스택 / 큐

 

basket 이라는 스택을 하나 만들고, moves 배열의 값을 차례대로 game 함수에 넘겨주면서 검사한다. 인형을 하나씩 뽑고 스택의 가장 위에 있는 인형과 비교하고 같으면 없애고 다르면 스택에 넣어준다.