[Algorithm/Swift] Codility / FrogRiverOne

문제

https://app.codility.com/programmers/lessons/4-counting_elements/frog_river_one/

풀이

개구리가 강을 건너기 위해서는 나뭇잎이 건너편까지 모두 강위에 떨어져야합니다.

주어진 배열의 인덱스는 초이며 배열의 요소 값은 위치 입니다.

Tip

• 중복을 허용하지 않는 Set을 이용해 값을 넣어주기

Code

public func solution(_ X : Int, _ A : inout [Int]) -> Int {
    guard !A.isEmpty else { return -1 }
    guard A.count >= X else { return -1 }
    
    var result = -1
    var leaves: Set<Int> = []
    
    for (index, i) in A.enumerated() {
        
        leaves.insert(i)
        if leaves.count == X {
            result = index
            break
        }
    }
    
    return result
}