6일차 - 수열과 구간 쿼리 2

5. 수열과 구간 쿼리 2

문제 설명
정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.
각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾습니다.
각 쿼리의 순서에 맞게 답을 저장한 배열을 반환하는 solution 함수를 완성해 주세요.
단, 특정 쿼리의 답이 존재하지 않으면 -1을 저장합니다.

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int[][] queries) {

        int[] answer = new int[queries.length];

        for (int i = 0; i < queries.length; i++) {
            int s = queries[i][0];
            int e = queries[i][1];
            int k = queries[i][2];

            int tmp = Integer.MAX_VALUE;

            for (int a = s; a <= e; a++) {
                if (arr[a] > k && arr[a] < tmp) {
                    tmp = arr[a];
                }
            }

            if (tmp == Integer.MAX_VALUE)
                answer[i] = -1;
            else
                answer[i] = tmp;
        }

        return answer;
    }
}

.

[링크: 수열과 구간 쿼리 2]