-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathCrane.py
More file actions
40 lines (30 loc) · 908 Bytes
/
Crane.py
File metadata and controls
40 lines (30 loc) · 908 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
문제링크 : https://programmers.co.kr/learn/courses/30/lessons/64061?language=python3
입력예시
board
[[0,0,0,0,0],[0,0,1,0,3],[0,2,5,0,1],[4,2,4,4,2],[3,5,1,3,1]]
moves
[1,5,3,5,1,2,1,4]
결과예시
4
"""
def solution(board, moves):
answer = 0
basket = []
for move in moves:
doll = selectDoll(move, board)
if (doll != 0):
basket.append(doll)
basket_size = len(basket)
if basket_size > 1 and basket[basket_size - 1] == basket[basket_size - 2]:
basket.pop()
basket.pop()
answer += 2
return answer
def selectDoll(move, board):
for line in range(len(board)):
if 0 != board[line][move - 1]:
doll = board[line][move - 1]
board[line][move - 1] = 0
return doll
return 0