Coding Practice/프로그래머스 Lv.0
수열과 구간 쿼리 2
ailen22
2024. 1. 26. 23:38
[문제]
정수 배열 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();
}
}