programmers

[JAVA] ★ 수열과 구간 쿼리2 Arrays.fill

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

풀이 1) 초기값 Arrays.fill(번수명,초기값)

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 i=0; i<queries.length; i++){
            int s =queries[i][0];
            int e =queries[i][1];
            int k =queries[i][2];
            int min = Integer.MAX_VALUE;
            for(int j=0; j<arr.length; j++){
                if(j>=s && j<=e && arr[j]>k){
                    if(min>=arr[j]){
                        min=arr[j];
                        answer[i]=arr[j]; 
                    }
                } 
            }
        }      
        return answer;
    }
}