크레인 인형뽑기 - 프로그래머스(Swift)

크레인 인형뽑기 - 프로그래머스(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/64061?language=swift 

프로그래머스

 

 나의 풀이

 

밑에서부터 탐색하기 위해 stride메서드를 사용했다

내 풀이에서는 먼저 결과배열에 넣고 중복되면 제거해서 해결했다

import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    var results:[Int] = []
    var array = board
    var deleteObj = 0
    for num in moves {
        var maxIdx = 0
        var selectedVal = 0
        for i in stride(from: board.count-1, through: 0, by: -1) {
            if array[i][num-1] > 0 {
                maxIdx = i
                selectedVal = array[i][num-1]
            }
        }
        array[maxIdx][num-1] = 0
        if selectedVal > 0 {
            results.append(selectedVal)    
        }
        if results.count > 1 {
            if results[results.count-1] == results[results.count-2] {
                deleteObj += 2
                results.removeLast()
                results.removeLast()
            }
        }
    }
    return deleteObj
}

 

 

 다른 사람의 풀이

 

결과배열에 값을 넣고 로직을 처리하는 내 풀이와 달리 넣기 전에 처리함으로써 더 간결하게 짤 수 있게 되었다.

배열을 사용할 때 last프로퍼티를 이용함으로써 맨 마지막 인덱스에 쉽게 접근이 가능했다.

그리고 배열을 위에서부터 탐색하되 값이 있으면 break함으로써 시간이 덜 걸리기 때문에 더 좋은 코드인 것 같다.

import Foundation

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