수열과 구간 쿼리 2

문제 설명

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.

각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾습니다.

각 쿼리의 순서에 맞게 답을 저장한 배열을 반환하는 solution 함수를 완성해 주세요.
단, 특정 쿼리의 답이 존재하지 않으면 -1을 저장합니다.

제한 사항

1 ≤ arr의 길이 ≤ 1,000
0 ≤ arr의 원소 ≤ 1,000,000
1 ≤ queries의 길이 ≤ 1,000
0 ≤ s ≤ e < arr의 길이
0 ≤ k ≤ 1,000,000

나의 코드

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 min = 1000001;
            for(int j=s; j<=e; j++) {
                if(arr[j]>k && arr[j]<min) {
                        min = arr[j];
                }
            }
            if(min != 1000001) {
                answer[i] = min;
            } else answer[i] = -1;
        }
        return answer;
    }
}

다른 사람 코드

import java.util.Arrays;

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

        int[] answer = new int[queries.length];
        Arrays.fill(answer, -1);

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

            for (int i = s; i <= e; i++) {
                if (k < arr[i]) {
                    answer[idx] = answer[idx] == -1 ? arr[i] : Math.min(answer[idx], arr[i]);
                }
            }

        }

        return answer;
    }
}

Arrays.fill 메소드, Math.min() 메소드 활용

느낀 점

범위야 정해주면 되는 거고, 아직 나오지 않는 최대값을 정하는 게 고민됐는데 제한사항의 범위를 참고했다. k로 나올 수 있는 값이 최대 1,000,000까지여서 min을 1,000,001로 두고 식을 짰다. 문제를 이해하는 데에 많은 시간이 소요됐던 문제였다.