[프로그래머스] 크레인 인형뽑기 게임

<문제>

---

 

<전략짜기>

ㅋㅋ 문제가 더길다.😂 문제에서 시키는대로 크레인의 위치를 찾고, 크레인을 쭉 내렸을때 집을 인형이 있다면 바구니에 옮겨서 제일 위에있는 인형과 같은 인형이면 두개를 터트리면 된다. 근데 문제는.. 주어지는 board를 그대로 사용하게 되면 인형을 집어올때나, 해당 라인이 비어있는지 체크하기가 좀 애매해진다. board[행][열]로 바로 접근하기가 좀.. 😭 그래서 나는 board를 좀 더 쉽게 보기위해 조금 회전시켜서 새로운 dolls로 사용하기로 했다. 아래처럼 돌리면, 해당라인이 비었는지와 제일 위에있는 인형을 가져오기 수월하기 때문이다!

좌측 (주어지는 board) / 우측 (회전시킨 board)

 

<코드>

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    var dolls = Array(repeating: Array(repeating: 0, count: board[0].count), count: board.count)
    var basket: [Int] = []
    var result = 0
    
    for (i, x) in board.enumerated() {
        for (j, y) in x.enumerated() {
            dolls[j][i] = y
        }
    }
    
    for move in moves {
        let realX = move-1
        let notEmptyCount = dolls[realX].filter { $0 != 0 }.count
        if notEmptyCount == 0 { continue }
        let realY = dolls[realX].count - notEmptyCount
        
        let curDoll = dolls[realX][realY]
        dolls[realX][realY] = 0
        
        if basket.count > 0 && curDoll == basket[basket.count-1] {
            result += 2
            basket.popLast()
        } else {
            basket.append(curDoll)
        }
    }
    return result
}

 

<Lesson Learned>

오랜만에 알고리즘 문제를 풀었다 ㅎㅎ 재밌군 👍 꾸준히 풀어야겠는걸!? 이번에 배운것중 하나는 배열에서 popLost()를 하면 마지막 element를 제거가 가능하다는 점이다. 원래는 array.remove(at: array.count-1) 를 사용해서 지웠었는데.. 굳이 그럴필요가 없었다! ㅋㅋ 다른 사람들은 board를 돌리지 않고 풀긴 했는데.. 아무래도 개인적으로는 이렇게 돌려서 쉽게 구현하는게 좋은 것 같다 👍

 

 

 

 

출저 : 프로그래머스 크레인 인형뽑기 게임