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
}
//
PREVIOUS54. 카운트 다운
NEXT56. 리스트 자르기