[Algorithm] 크레인 인형뽑기 게임 (append(), removeLast())

🖥 문제

게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.

🤔 풀이 방법

1. 변수
   • copiedBoard : board의 내용을 복사해온다.
   • basket : 인형을 담을 바구니
   • doll : 집어 올린 인형
   • result : 터트려져 사라진 인형의 개수
2. 2중 for문
   • 크레인을 작동시킨 위치를 하나씩 탐색한다.
   • 동시에 화면의 격자 상태가 상태가 담긴 2차원 배열 board를 탐색한다.
   • for move in moves { for i in 0..<board.count { } }
3. doll = copiedBoard[i][move - 1]
   • 인형을 잡은 결과를 doll에 저장한다.
4. if doll != 0 : 인형을 잡았을 경우
   • copiedBoard[i][move - 1] = 0
   • 인형을 잡은 위치에 0을 넣어준다.
5. basket.append(doll) : 바구니에 인형을 넣는다.
   • 바구니가 비어있을 경우
   • 바구니에 들어있는 마지막 인형이 잡은 인형과 다를 경우
6. basket.removeLast() : 바구니에 들어있는 마지막 인형을 터트린다.
   • 바구니에 들어있는 마지막 인형과 잡은 인형이 같을 경우
   • 잡은 인형은 터트려져 바구니에 들어가지 않는다.
   • result += 2로 터트려져 없어진 인형의 수를 올려준다.

👩🏻‍💻 Swift Code

import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    var copiedBoard = board
    var basket: [Int] = []
    var doll: Int = 0
    var result: Int = 0

    for move in moves {
        for i in 0..<board.count {
            doll = copiedBoard[i][move - 1]
            if doll != 0 {
                copiedBoard[i][move - 1] = 0
                if basket.count > 0 && basket.last == doll {
                    basket.removeLast()
                    result += 2
                    break
                } else {
                    basket.append(doll)
                    break
                }
            }

        }
    }
    return result
}

 

코딩테스트 연습 - 크레인 인형뽑기 게임