알고리즘/구현
파이썬 enumerate() 내장함수 사용
mimi04
2024. 1. 12. 14:37
*처음 enumerate() 함수 쓰지 않고 푼 내 코드
def solution(board, moves):
answer = 0
bag = []
for i in moves:
for j in range(len(board)):
if board[j][i-1] != 0:
if bag and bag[-1] == board[j][i-1]:
del bag[-1]
answer += 2
else:
bag.append(board[j][i-1])
board[j][i-1] = 0
break
return answer
* enumerate() 함수 사용한 코드
def solution(board, moves):
answer = 0
bag = []
for i in moves:
for index, j in enumerate(board):
if j[i-1] != 0:
if bag and bag[-1] == j[i-1]:
del bag[-1]
answer += 2
else:
bag.append(j[i-1])
board[index][i-1] = 0
break
return answer
반응형