55. 가까운 1 찾기

import Foundation

func solution(_ arr:[Int], _ idx:Int) -> Int {

    var answer : Int = 0


    for i in idx ... arr.count-1 {
        if arr[i] == 1 {

            answer = i
            break

        } else {

            answer = -1

        }
    }

    return answer
}

이렇게 한 이유

반복문을 통하려고 하였고, 1일때의 인덱스값을 찾자마자 리턴을 시켜야 했기에, break를 써서 끊었다.

그리고 그것이 해당하지 않는 케이스라면 -1을 리턴을 하여 해결하였다.

다른 코드를 보니 엄청 심플하게 했다…

다른 코드

import Foundation

func solution(_ arr:[Int], _ idx:Int) -> Int {
    return arr[idx...].firstIndex(of: 1) ?? -1
}

//