[Algorithm] 프로그래머스 - 가까운 1 찾기 (Kotlin)

정수 배열 arr이 주어진다. arr의 원소는 1 또는 0이다. 정수 idx가 주어졌을 때 idx보다 크면서 배열의 값이 1인 가장 작은 인덱스를 찾아 리턴하는 함수를 완성하라. 단 그러한 인덱스가 없다면 -1을 리턴한다

 

 

구현 단계에서 막혀 풀지 못해 다른 사람의 풀이를 참고했다.

 

class Solution {
    fun solution(arr: IntArray, idx: Int): Int {
        var answer = 0
        for (i in arr.indices) {
            if (i >= idx && arr[i] == 1) {
                answer = i
                break
            } else {
                answer = -1
            }
        }
        return answer
    }
}