[Algorithm] 프로그래머스 - 수열과 구간 쿼리 2 (Kotlin)

 

수열과 구간 쿼리 3보다 좀 더 꼬아진 문제지만 2차원 배열 안의 요소에 접근한다는 큰 흐름은 이전 문제와 동일하기 때문에 일단 주먹구구식으로 풀었다

 

class Solution {
    fun solution(arr: IntArray, queries: Array<IntArray>): IntArray {
        val answer = IntArray(queries.size)
        for (i in queries.indices) {
            val s = queries[i][0]
            val e = queries[i][1]
            val k = queries[i][2]
            var min = Int.MAX_VALUE
            for (j in s..e) {
                if (arr[j] > k && arr[j] < min) {
                    min = arr[j]
                }
            }
            answer[i] = if (min == Int.MAX_VALUE) -1 else min
        }
        return answer
    }
}