[프로그래머스] 수열과 구간 쿼리 2 자바 java

https://school.programmers.co.kr/learn/courses/30/lessons/181923

프로그래머스

문제 설명
정수 배열 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

 

---

문제를 간단하게 풀어보면 

1. arr이라는 정수 배열이 주어지고 queris라는 [s,e,k] 꼴의 배열이 주어진다. 

2. 각 s와 e는 정수배열의 인덱스(i)를 의미하고  s<=i<=e의 범위를 나타낸다. 

3. arr에서 범위 내의 정수 중 정수 k보다 큰 값 중 가장 작은 값을 넣어  결과 배열을 반환하라. 

 

그렇다면 쿼리의 길이만큼 반복하면서 주어진 범위를 돌며 가장 작은 값을 구하는 코드를 작성하면 된다. 

가장 작은 값을 구해야 하기 때문에 결과 배열을 Integer.MAX_VALUE로 모두 초기화하였고, 

범위 안에서 값을 찾지 못해 여전히 결과 배열의 값이 Integer.MAX_VALUE인 경우에는 -1로 값을 바꾸어 주도록 코드를 작성했다.  

 

전체코드

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] ans = new int[queries.length];
        Arrays.fill(ans,Integer.MAX_VALUE);
        for(int i=0; i<queries.length;i++){
            int s = queries[i][0];
            int e = queries[i][1];
            int k = queries[i][2];
            
            for(int j=s;j<=e;j++){
                if(arr[j]>k){
                    ans[i]=Math.min(arr[j],ans[i]);
                }
            }
            if(ans[i]==Integer.MAX_VALUE)
                ans[i]=-1;
        }
        return ans;
    }
}