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

프로그래머스

---

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

나의 풀이

import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    var array = board
    var basket = [Int]()
    var count = 0
    for i in moves {
        for j in (0..<array[0].count) {
            if array[j][i-1] != 0 {
                basket.append(array[j][i-1])
                array[j][i-1] = 0
                break
            }
        }
        if basket.count >= 2 && basket[basket.endIndex-1] == basket[basket.endIndex-2] {
            basket.removeLast()
            basket.removeLast()
            count += 2
        }
    }
    return count
}

 

---

개선된 풀이

import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    var array = board
    var basket = [Int]()
    var count = 0
    for i in moves {
        for j in (0..<array[0].count) {
            if array[j][i-1] != 0 {
                
                let doll = array[j][i-1]
                array[j][i-1] = 0
                
                if basket.last == doll { basket.removeLast(); count += 2 }
                else { basket.append(doll) }
                
                break
            }
        }
    }
    return count
}

 

첫 번째 풀이는

일단 배열에 인형을 넣어준 다음

마지막 인형과 그전에 있는 인형이 같으면

두 인형을 없어지게 했었는데요.

 

차라리

먼저 같은지 비교를 하고

다르면 인형을 넣어주고

같으면 원래 배열에 있는 마지막 인형만 지워주는 식으로

불필요하게 넣었다 다시 빼는 일을 줄였습니다.

 

하지만 왜 첫 번째 풀이 코드가 시간은 더 빠를까요..? 🥲

 

좌: 첫번째 풀이 / 우: 두번째 풀이