코린이의 소소한 공부노트

[프로그래머스/Lv.0] 수열과 구간 쿼리 2 본문

코딩테스트 풀이/JAVA

[프로그래머스/Lv.0] 수열과 구간 쿼리 2

무지맘 2023. 5. 1. 19:45

1. Input, Output, Example

queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴이다. query마다 순서대로 s i e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾는다.

- 각 쿼리의 순서에 맞게 답을 저장한 배열을 반환

 

2. Constraint

1) 1 arr의 길이 1,000

2) 0 arr의 원소 1,000,000

3) 1 queries의 길이 1,000

4) 0 s e < arr의 길이

5) 0 k 1,000,000

 

3. Code

1) 첫 코드(2023/05/01)

class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = new int[queries.length];
        for(int i=0 ; i<queries.length ; i++){
            int min = Integer.MAX_VALUE;
            boolean find = false;
            for(int j=queries[i][0] ; j<=queries[i][1] ; j++)
                if(arr[j]>queries[i][2] && arr[j]<min){
                    find = true; min = arr[j];
                }
            if(find) answer[i] = min;
            else answer[i] = -1;
        }
        return answer;
    }
}