수열과 구간 쿼리 2

[문제]

정수 배열 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]

 

 

 

 

 

 

[답]

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = new int[queries.length];
        //answr값과 가장 큰 값으로 배열 초기화
        Arrays.fill(answer, Integer.MAX_VALUE);

        for(int i = 0; i < queries.length; i++) {
        	//이해못한 부분(다시공부)
            for(int j = queries[i][0]; j <= queries[i][1]; j++) {
                if(arr[j] > queries[i][2]) {
                    answer[i] = Math.min(answer[i], arr[j]);
                }
            }
			
            if(answer[i] == Integer.MAX_VALUE) {
                answer[i] = -1;
            }
        }
        return answer;
    }
}

 

Arrays.fill() : 배열을 지정한 값으로 모두 초기화시킴, 특정값으로 배열 채우기

Math.min() : 둘 중 작은값 반환

 

 

 

 

[다른사람 풀이]

import java.util.Arrays;
class Solution1 {
    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;
    }
}




import java.util.stream.IntStream;
class Solution {
    public int[] solution(int[] arr, int[][] queries) {
        int[] answer = {};
        //Stream사용(아직 공부중)
        return IntStream.range(0, queries.length)
                .map(q -> IntStream.rangeClosed(queries[q][0], queries[q][1])
                        .map(i -> arr[i])
                        .filter(i -> i > queries[q][2])
                        .min().orElse(-1)
                ).toArray();
    }
}