본문 바로가기
코딩테스트 연습

2019 카카오 개발자 겨울 인턴십_크레인 인형뽑기

by 부자독개 2020. 4. 17.
반응형

def solution(board, moves):
    answer = 0
    box = [] ## 뽑은 인형을 담을 박스 
    for i in moves: ## moves만큼 뽑기
      for k in range(len(board)): ## board의 행의 길이가 탐색해야할 깊이

        if board[k][i-1] == 0 : pass ## 인형이 없으면 지나가기
        else: ## 인형이 있으면 박스에 옮기고 0(비워진 상태)로 바꾸기. 그리고 다음 뽑기를 위해 반복문 탈출.
          box.append(board[k][i-1])
          board[k][i-1] = 0
          break


      if len(box)>=2:  ## 박스에 인형이 2개 이상일때 체크
        for k in range(len(box)-1):
          if box[k] == box[k+1] :  ## 같은 인형이 있으면 answer를 2씩 증가, 인형두개 터트리기.
            answer += 2  
            del box[k]
            del box[k]
    return answer

반응형

댓글